Print more infomation

d171c6ab · rocking · df31025f · d171c6ab
Commit d171c6ab authored Jan 18, 2023 by rocking
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

include/ck/tensor_operation/gpu/device/impl/device_gemm_multiple_d_layernorm_xdl_cshuffle.hpp ...ce/impl/device_gemm_multiple_d_layernorm_xdl_cshuffle.hpp +6 -1

No files found.
--- a/include/ck/tensor_operation/gpu/device/impl/device_gemm_multiple_d_layernorm_xdl_cshuffle.hpp
+++ b/include/ck/tensor_operation/gpu/device/impl/device_gemm_multiple_d_layernorm_xdl_cshuffle.hpp
@@ -1062,7 +1062,12 @@ struct DeviceGemmMultipleDLayernorm_Xdl_CShuffle
            << GemmKPerBlock << ", "
            << AK1 << ", "
            << BK1 << ", "
-            << getGemmSpecializationString(GemmSpec)
+            << getGemmSpecializationString(GemmSpec) << ", "
+            << PostShuffleThreadClusterSize_M_N::At(I0) << ", "
+            << PostShuffleThreadClusterSize_M_N::At(I1) << ", "
+            << LayernormThreadClusterSize_M_N::At(I0) << ", "
+            << LayernormThreadClusterSize_M_N::At(I1) << ", "
+            << LayernormThreadSliceSize_M
            << ">"
            << " LoopScheduler: "
            << LoopSchedToString[LoopSched] << ", "