From 52b16da8d5f7ae90d3a6e7d9f1e7a8b7ca3fccb0 Mon Sep 17 00:00:00 2001 From: mtgu0705 Date: Thu, 5 Jun 2025 22:10:05 -0500 Subject: [PATCH] added mxfp4 moe gemm2 example, function passed --- example/67_gemm_microscaling/moe_gemm1_xdl_mx_fp4_bns.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/example/67_gemm_microscaling/moe_gemm1_xdl_mx_fp4_bns.cpp b/example/67_gemm_microscaling/moe_gemm1_xdl_mx_fp4_bns.cpp index 0881f74cc2..ce8e02912c 100644 --- a/example/67_gemm_microscaling/moe_gemm1_xdl_mx_fp4_bns.cpp +++ b/example/67_gemm_microscaling/moe_gemm1_xdl_mx_fp4_bns.cpp @@ -162,8 +162,8 @@ using DeviceOpInstance = ck::tensor_operation::device::Devic 16, 16, 16, 16, 4, 2, - 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, + S<8, 32, 1>, S<1, 0, 2>, S<1, 0, 2>, 2, 16, 16, 1, + S<8, 32, 1>, S<1, 0, 2>, S<1, 0, 2>, 2, 16, 16, 1, 2, 2, S<1, 32, 1, 8>, S<8, 1, 1, 1>, ck::BlockGemmPipelineScheduler::Intrawave, ck::BlockGemmPipelineVersion::v3, ActOP, Nswizzle, true, MulRoutedWeight, ck::index_t, A0DataType>;