add coment

0a9ccef6 · Jing Zhang · root · d9cf9ec0 · 0a9ccef6
Commit 0a9ccef6 authored Jun 30, 2023 by Jing Zhang Committed by root Jun 30, 2023
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

include/ck/tensor_operation/gpu/device/impl/device_grouped_gemm_xdl_splitk_cshuffle.hpp ...u/device/impl/device_grouped_gemm_xdl_splitk_cshuffle.hpp +2 -1

No files found.
--- a/include/ck/tensor_operation/gpu/device/impl/device_grouped_gemm_xdl_splitk_cshuffle.hpp
+++ b/include/ck/tensor_operation/gpu/device/impl/device_grouped_gemm_xdl_splitk_cshuffle.hpp
@@ -144,7 +144,7 @@ template <typename ALayout,
          typename DsDataType,
          typename EDataType,
          typename AElementwiseOperation,
-          typename BElementwiseOperation,
+          ypename BElementwiseOperation,
          typename CDEElementwiseOperation,
          GemmSpecialization GemmSpec,
          ck::index_t NumGemmKPrefetchStage,
@@ -415,6 +415,7 @@ struct DeviceGroupedGemmXdlSplitKCShuffle : public DeviceGroupedGemmSplitK<ALayo
            index_t StrideB;
            index_t StrideC;

+	    //do not need after loop M implemented
            index_t block_start_;
            index_t block_end_;
        };