Fix stride error

ea6b9481 · rocking · 2b5ec663 · ea6b9481
Commit ea6b9481 authored Jan 28, 2022 by rocking
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

device_operation/include/device_gemm_xdl_c_shuffle_bias.hpp device_operation/include/device_gemm_xdl_c_shuffle_bias.hpp +2 -2

No files found.
--- a/device_operation/include/device_gemm_xdl_c_shuffle_bias.hpp
+++ b/device_operation/include/device_gemm_xdl_c_shuffle_bias.hpp
@@ -132,8 +132,8 @@ struct DeviceGemmXdl_C_Shuffle_Bias
    static auto MakeC0GridDescriptor_M_N(index_t M, index_t N)
    {
-        // TODO - bias with stride (I1, I0)
+        // TODO - bias with stride (I0, I1)
-        return make_naive_tensor_descriptor(make_tuple(M, N), make_tuple(I0, I1));
+        return make_naive_tensor_descriptor(make_tuple(M, N), make_tuple(I1, I0));
    }
    using AGridDesc_K0_M_K1 = decltype(MakeAGridDescriptor_K0_M_K1(1, 1, 1));