From 6034a060b5ec8079a4090bda73ce464da0fd8a1f Mon Sep 17 00:00:00 2001 From: mtgu0705 Date: Mon, 30 Jun 2025 01:43:32 -0500 Subject: [PATCH] change the bns gemm2 kernel --- example/67_gemm_microscaling/moe_gemm2_xdl_mx_fp4_bns.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/example/67_gemm_microscaling/moe_gemm2_xdl_mx_fp4_bns.cpp b/example/67_gemm_microscaling/moe_gemm2_xdl_mx_fp4_bns.cpp index 6718581a50..829bf9af24 100644 --- a/example/67_gemm_microscaling/moe_gemm2_xdl_mx_fp4_bns.cpp +++ b/example/67_gemm_microscaling/moe_gemm2_xdl_mx_fp4_bns.cpp @@ -158,7 +158,7 @@ using DeviceOpInstance = ck::tensor_operation::device::Devic 4, 4, S<8, 32, 1>, S<1, 0, 2>, S<1, 0, 2>, 2, 16, 16, 0, S<8, 32, 1>, S<1, 0, 2>, S<1, 0, 2>, 2, 16, 16, 0, - 2, 2, S<1, 32, 1, 8>, S<2, 1, 1, 1>, + 2, 4, S<1, 4, 1, 64>, S<2, 1, 1, 1>, ck::BlockGemmPipelineScheduler::Intrawave, ck::BlockGemmPipelineVersion::v1, 0, false, false, MulRoutedWeight, ck::index_t, A0DataType>; // clang-format on