From 006e01bb9ab91a6cb120bfcc335653c54224ee94 Mon Sep 17 00:00:00 2001 From: coderfeli Date: Mon, 10 Mar 2025 10:10:50 +0000 Subject: [PATCH] fix typo --- include/ck/tensor_operation/gpu/grid/gridwise_moe_gemm.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/include/ck/tensor_operation/gpu/grid/gridwise_moe_gemm.hpp b/include/ck/tensor_operation/gpu/grid/gridwise_moe_gemm.hpp index 03fdcf5574..33701e4153 100644 --- a/include/ck/tensor_operation/gpu/grid/gridwise_moe_gemm.hpp +++ b/include/ck/tensor_operation/gpu/grid/gridwise_moe_gemm.hpp @@ -1244,7 +1244,7 @@ struct GridwiseMoeGemm // dummy constexpr auto b_block_desc_bk0_n_bk1 = GetBBlockDescriptor_BK0PerBlock_NPerBlock_BK1(); // A matrix blockwise copy - auto a_blockwise_copy = ThreadGroupTensorSliceTransfer_v4r1_gather< + auto a_blockwise_copy = ThreadGroupTensorSliceTransfer_v4r1_mod8< ThisThreadBlock, AElementwiseOperation, ck::tensor_operation::element_wise::PassThrough, @@ -1752,7 +1752,7 @@ struct GridwiseMoeGemm // dummy constexpr auto b_block_desc_bk0_n_bk1 = GetBBlockDescriptor_BK0PerBlock_NPerBlock_BK1(); // A matrix blockwise copy - auto a_blockwise_copy = ThreadGroupTensorSliceTransfer_v4r1_gather< + auto a_blockwise_copy = ThreadGroupTensorSliceTransfer_v4r1_mod8< ThisThreadBlock, AElementwiseOperation, ck::tensor_operation::element_wise::PassThrough,