diff --git a/include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma_cshuffle_v3.hpp b/include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma_cshuffle_v3.hpp index d5fc86b9e8..d37eebaed2 100644 --- a/include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma_cshuffle_v3.hpp +++ b/include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma_cshuffle_v3.hpp @@ -362,11 +362,11 @@ struct GridwiseGemm_wmma_cshuffle_v3 std::cout << "problem {" << "M:" << M << ", " << "N:" << N << ", " << "K:" << K << ", " << "SAs: {"; static_for<0, NumATensor, 1>{}([&](auto i) { - std::cout << StrideAs[i] << (i.value < NumDTensor - 1 ? ", " : ""); + std::cout << StrideAs[i] << (i.value < NumATensor - 1 ? ", " : ""); }); std::cout << "}, " << "SBs: {"; - static_for<0, NumATensor, 1>{}([&](auto i) { - std::cout << StrideBs[i] << (i.value < NumDTensor - 1 ? ", " : ""); + static_for<0, NumBTensor, 1>{}([&](auto i) { + std::cout << StrideBs[i] << (i.value < NumBTensor - 1 ? ", " : ""); }); std::cout << "}, "; if constexpr(NumDTensor > 0) diff --git a/include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma_cshuffle_v3_b_scale.hpp b/include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma_cshuffle_v3_b_scale.hpp index aeeda14235..46de6b156a 100644 --- a/include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma_cshuffle_v3_b_scale.hpp +++ b/include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma_cshuffle_v3_b_scale.hpp @@ -259,11 +259,11 @@ struct GridwiseGemm_wmma_cshuffle_v3_b_scale std::cout << "problem {" << "M:" << M << ", " << "N:" << N << ", " << "K:" << K << ", " << "SAs: {"; static_for<0, NumATensor, 1>{}([&](auto i) { - std::cout << StrideAs[i] << (i.value < NumDTensor - 1 ? ", " : ""); + std::cout << StrideAs[i] << (i.value < NumATensor - 1 ? ", " : ""); }); std::cout << "}, " << "SBs: {"; - static_for<0, NumATensor, 1>{}([&](auto i) { - std::cout << StrideBs[i] << (i.value < NumDTensor - 1 ? ", " : ""); + static_for<0, NumBTensor, 1>{}([&](auto i) { + std::cout << StrideBs[i] << (i.value < NumBTensor - 1 ? ", " : ""); }); std::cout << "}, "; if constexpr(NumDTensor > 0)