ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-30 03:37:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix format

Browse Source
This commit is contained in:
lalala-sh
2026-01-22 03:06:40 +00:00
parent 4b84f2c38b
commit 8d32b38fbd
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
include/ck/tensor_operation/gpu/device/impl/device_moe_gemm_blockscale.hpp
View File

@@ -256,8 +256,8 @@ struct DeviceMoeGemmBlockScale
if(arg_.KBatch > 1)
hipGetErrorString(hipMemsetAsync(arg_.p_c_grid,
0,
arg_.M * arg_.N * sizeof(CDataType)
* (IsInputGemm && IsSplitK ? 2 : 1),
arg_.M * arg_.N * sizeof(CDataType) *
(IsInputGemm && IsSplitK ? 2 : 1),
stream_config.stream_id_));
};
@@ -276,7 +276,7 @@ struct DeviceMoeGemmBlockScale
hipGetErrorString(hipMemsetAsync(arg.p_c_grid,
0,
arg.M * arg.N * sizeof(CDataType) *
(IsInputGemm && IsSplitK ? 2 : 1),
(IsInputGemm && IsSplitK ? 2 : 1),
stream_config.stream_id_));
ave_time = launch_and_time_kernel(