From 6e3df975a2ca2a8c7bd3261f4a1d2042f8a26d26 Mon Sep 17 00:00:00 2001 From: xws117 Date: Thu, 1 Feb 2024 10:41:30 +0800 Subject: [PATCH] Modify comments in code examples/08_turing_tensorop_gemm/turing_tensorop_gemm.cu (#1325) --- examples/08_turing_tensorop_gemm/turing_tensorop_gemm.cu | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/08_turing_tensorop_gemm/turing_tensorop_gemm.cu b/examples/08_turing_tensorop_gemm/turing_tensorop_gemm.cu index 844bd0fb..a24fd0e4 100644 --- a/examples/08_turing_tensorop_gemm/turing_tensorop_gemm.cu +++ b/examples/08_turing_tensorop_gemm/turing_tensorop_gemm.cu @@ -140,8 +140,8 @@ using ElementInputA = int8_t; // <- data type of elements using ElementInputB = int8_t; // <- data type of elements in input matrix B using ElementOutput = int32_t; // <- data type of elements in output matrix D -// The code section below describes matrix layout of input and output matrices. Column Major for -// Matrix A, Row Major for Matrix B and Row Major for Matrix C +// The code section below describes matrix layout of input and output matrices. Row Major for +// Matrix A, Column Major for Matrix B and Row Major for Matrix C using LayoutInputA = cutlass::layout::RowMajor; using LayoutInputB = cutlass::layout::ColumnMajor; using LayoutOutput = cutlass::layout::RowMajor;