fixed a bug in the gpu gemm implementation.

f140f4c6 · Shucai Xiao · 44732d6c · f140f4c6
Commit f140f4c6 authored Apr 04, 2019 by Shucai Xiao
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/targets/gpu/gemm.cpp src/targets/gpu/gemm.cpp +1 -1

No files found.
--- a/src/targets/gpu/gemm.cpp
+++ b/src/targets/gpu/gemm.cpp
@@ -342,7 +342,7 @@ argument miopen_gemm::compute(context& ctx,
            bool transb       = args[1].get_shape().transposed();
            rocblas_int lda   = args[0].get_shape().strides()[transa ? dim_1 : dim_0];
            rocblas_int ldb   = args[1].get_shape().strides()[transb ? dim_1 : dim_0];
-            rocblas_int ldc   = args[3].get_shape().strides()[dim_0];
+            rocblas_int ldc   = args[2].get_shape().strides()[dim_0];
            auto out_lens     = output_shape.lens();
            rocblas_int m     = out_lens[dim_0];
            rocblas_int n     = out_lens[dim_1];