From c9591a694d683df7778504749cacdc71fd8b3b3e Mon Sep 17 00:00:00 2001 From: Jee Li Date: Fri, 5 Jan 2024 01:41:39 +0800 Subject: [PATCH] fix typo (#1279) --- examples/12_gemm_bias_relu/gemm_bias_relu.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/12_gemm_bias_relu/gemm_bias_relu.cu b/examples/12_gemm_bias_relu/gemm_bias_relu.cu index 418540f399..cbe6eb39fa 100644 --- a/examples/12_gemm_bias_relu/gemm_bias_relu.cu +++ b/examples/12_gemm_bias_relu/gemm_bias_relu.cu @@ -81,7 +81,7 @@ using ShapeMMAThreadBlock = // This code section describes tile size a warp will compute using ShapeMMAWarp = cutlass::gemm::GemmShape<64, 64, 32>; // <- warp tile M = 64, N = 64, K = 32 // This code section describes the size of MMA op -using ShapeMMAOp = cutlass::gemm::GemmShape<16, 8, 8>; // <- MMA Op tile M = 8, N = 8, K = 4 +using ShapeMMAOp = cutlass::gemm::GemmShape<16, 8, 8>; // <- MMA Op tile M = 16, N = 8, K = 8 // This code section describes how threadblocks are scheduled on GPU using SwizzleThreadBlock = cutlass::gemm::threadblock::GemmIdentityThreadblockSwizzle<>; // <- ??