From e48c7618e4d06b8ff6169cd78de5826019a241b5 Mon Sep 17 00:00:00 2001 From: LiYu Lu Date: Wed, 10 Jul 2024 23:12:36 +0800 Subject: [PATCH] [bug] fix device thread `gemm.h` constructor (#1473) --- tools/util/include/cutlass/util/reference/device/thread/gemm.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/util/include/cutlass/util/reference/device/thread/gemm.h b/tools/util/include/cutlass/util/reference/device/thread/gemm.h index 1bc589fd..04775a74 100644 --- a/tools/util/include/cutlass/util/reference/device/thread/gemm.h +++ b/tools/util/include/cutlass/util/reference/device/thread/gemm.h @@ -91,7 +91,7 @@ struct Gemm { A_tile[i] = ElementA(0); } - for (int j = 0; j < OutputTile::kColumn; ++j) { + for (int j = 0; j < OutputTile::kRow; ++j) { B_tile[j] = ElementB(0); }