update

7b1ad58a · Chao Liu · d49d6928 · 7b1ad58a
Commit 7b1ad58a authored Jul 05, 2022 by Chao Liu
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

include/ck/tensor_operation/gpu/device/device_contraction_multiple_d_xdl_cshuffle.hpp ...gpu/device/device_contraction_multiple_d_xdl_cshuffle.hpp +6 -1

No files found.
--- a/include/ck/tensor_operation/gpu/device/device_contraction_multiple_d_xdl_cshuffle.hpp
+++ b/include/ck/tensor_operation/gpu/device/device_contraction_multiple_d_xdl_cshuffle.hpp
@@ -958,12 +958,17 @@ struct DeviceContractionMultipleD_Xdl_CShuffle
        // clang-format off
        str << "DeviceContractionMultipleD_Xdl_CShuffle"
            << "<"
+            << NumDimM << ", "
+            << NumDimN << ", "
+            << NumDimK << ", "
            << BlockSize << ", "
            << MPerBlock << ", "
            << NPerBlock << ", "
            << KPerBlock << ", "
            << AK1 << ", "
-            << BK1
+            << BK1 << ", "
+            << ABlockTransferSrcVectorDim << ", "
+            << BBlockTransferSrcVectorDim
            << ">";
        // clang-format on