From 6407bcdf0a24097b7b016ee105937693c62f9923 Mon Sep 17 00:00:00 2001 From: Yujia Zhai Date: Tue, 12 Sep 2023 11:04:18 -0700 Subject: [PATCH] fix matrix B indices (#1089) --- .../48_hopper_warp_specialized_gemm.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/48_hopper_warp_specialized_gemm/48_hopper_warp_specialized_gemm.cu b/examples/48_hopper_warp_specialized_gemm/48_hopper_warp_specialized_gemm.cu index 194c4005..5a98c7ae 100644 --- a/examples/48_hopper_warp_specialized_gemm/48_hopper_warp_specialized_gemm.cu +++ b/examples/48_hopper_warp_specialized_gemm/48_hopper_warp_specialized_gemm.cu @@ -324,7 +324,7 @@ typename Gemm::Arguments args_from_options(const Options &options) bool verify(const Options &options) { cutlass::TensorRef ref_A(block_A.get(), Gemm::LayoutA::packed({options.m, options.k})); - cutlass::TensorRef ref_B(block_B.get(), Gemm::LayoutB::packed({options.n, options.k})); + cutlass::TensorRef ref_B(block_B.get(), Gemm::LayoutB::packed({options.k, options.n})); cutlass::TensorRef ref_C(block_C.get(), Gemm::LayoutC::packed({options.m, options.n})); cutlass::TensorRef ref_D(block_ref_D.get(), Gemm::LayoutD::packed({options.m, options.n}));