ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-03 05:01:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle moe_fp8 no-mainloop cases. Supprese no-mainloop check (#2438)

Browse Source 
Co-authored-by: felix <felix.li@amd.com>
This commit is contained in:
huaiguxu
2025-07-16 15:44:34 +08:00
committed by GitHub
parent 3499fe67ff
commit c1badfd30c
2 changed files with 45 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/ck/tensor_operation/gpu/grid/gridwise_moe_gemm.hpp
View File

@@ -1112,7 +1112,7 @@ struct GridwiseMoeGemm
}
// check gridwise gemm pipeline
#if 1
#if 0
const auto num_k_loop = karg.AK0 / (KPerBlock / AK1Value);
if(num_k_loop <= BlockwiseGemmPipe::PrefetchStages)