Skip to content

GitLab

Commit 0a9ccef6 authored by Jing Zhang's avatar Jing Zhang Committed by root
Browse files

add coment

parent d9cf9ec0
Hide 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
View file @ 0a9ccef6
...@@ -144,7 +144,7 @@ template <typename ALayout, ...@@ -144,7 +144,7 @@ template <typename ALayout,
typename DsDataType, typename DsDataType,
typename EDataType, typename EDataType,
typename AElementwiseOperation, typename AElementwiseOperation,
typename BElementwiseOperation, ypename BElementwiseOperation,
typename CDEElementwiseOperation, typename CDEElementwiseOperation,
GemmSpecialization GemmSpec, GemmSpecialization GemmSpec,
ck::index_t NumGemmKPrefetchStage, ck::index_t NumGemmKPrefetchStage,
...@@ -415,6 +415,7 @@ struct DeviceGroupedGemmXdlSplitKCShuffle : public DeviceGroupedGemmSplitK<ALayo ...@@ -415,6 +415,7 @@ struct DeviceGroupedGemmXdlSplitKCShuffle : public DeviceGroupedGemmSplitK<ALayo
index_t StrideB; index_t StrideB;
index_t StrideC; index_t StrideC;
//do not need after loop M implemented
index_t block_start_; index_t block_start_;
index_t block_end_; index_t block_end_;
}; };
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment