Include # prefetch in type string of DeviceGemm_Xdl_CShuffle<>

a86fc096 · Po-Yen, Chen · 353a3e00 · a86fc096
Commit a86fc096 authored Jul 20, 2023 by Po-Yen, Chen
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

include/ck/tensor_operation/gpu/device/impl/device_gemm_xdl_cshuffle.hpp ...or_operation/gpu/device/impl/device_gemm_xdl_cshuffle.hpp +1 -0

No files found.
--- a/include/ck/tensor_operation/gpu/device/impl/device_gemm_xdl_cshuffle.hpp
+++ b/include/ck/tensor_operation/gpu/device/impl/device_gemm_xdl_cshuffle.hpp
@@ -275,6 +275,7 @@ struct DeviceGemm_Xdl_CShuffle : public DeviceGemm<ALayout,
        // clang-format off
        str << "DeviceGemm_Xdl_CShuffle"
            << "<"
+            << NumGemmKPrefetchStage << ", "
            << BlockSize << ", "
            << MPerBlock << ", "
            << NPerBlock << ", "