fixed issue164 (#165)

* fixed issue164 * removed prints

fixed issue164 (#165)
* fixed issue164 * removed prints
ecf337ba · Jianfeng Yan · GitHub · f015c776 · ecf337ba
Unverified Commit ecf337ba authored Mar 31, 2022 by Jianfeng Yan Committed by GitHub Mar 31, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

include/ck/tensor_operation/gpu/device/device_batched_gemm_xdl.hpp ...k/tensor_operation/gpu/device/device_batched_gemm_xdl.hpp +3 -3

No files found.
--- a/include/ck/tensor_operation/gpu/device/device_batched_gemm_xdl.hpp
+++ b/include/ck/tensor_operation/gpu/device/device_batched_gemm_xdl.hpp
@@ -129,7 +129,7 @@ struct DeviceBatchedGemmXdl
            }
            else if constexpr(is_same<tensor_layout::gemm::ColumnMajor, ALayout>::value)
            {
-                return make_naive_tensor_descriptor(make_tuple(M, K), make_tuple(I1, M));
+                return make_naive_tensor_descriptor(make_tuple(M, K), make_tuple(I1, StrideA));
            }
        }();

@@ -158,7 +158,7 @@ struct DeviceBatchedGemmXdl
            }
            else if constexpr(is_same<tensor_layout::gemm::ColumnMajor, BLayout>::value)
            {
-                return make_naive_tensor_descriptor(make_tuple(K, N), make_tuple(I1, K));
+                return make_naive_tensor_descriptor(make_tuple(K, N), make_tuple(I1, StrideB));
            }
        }();

@@ -183,7 +183,7 @@ struct DeviceBatchedGemmXdl
            }
            else if constexpr(is_same<tensor_layout::gemm::ColumnMajor, CLayout>::value)
            {
-                return make_naive_tensor_descriptor(make_tuple(M, N), make_tuple(I1, M));
+                return make_naive_tensor_descriptor(make_tuple(M, N), make_tuple(I1, StrideC));
            }
        }();