From 4000df9567cfe409c6333627a48cadce0237621c Mon Sep 17 00:00:00 2001 From: xue-fc <43108691+xue-fc@users.noreply.github.com> Date: Thu, 22 Apr 2021 20:28:28 +0800 Subject: [PATCH] fix a wrong description --- .../ampere_sparse_tensorop_gemm.cu | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/15_ampere_sparse_tensorop_gemm/ampere_sparse_tensorop_gemm.cu b/examples/15_ampere_sparse_tensorop_gemm/ampere_sparse_tensorop_gemm.cu index f8c300f2..5034f454 100644 --- a/examples/15_ampere_sparse_tensorop_gemm/ampere_sparse_tensorop_gemm.cu +++ b/examples/15_ampere_sparse_tensorop_gemm/ampere_sparse_tensorop_gemm.cu @@ -57,8 +57,8 @@ using ElementInputA = cutlass::int4b_t; // <- data type of elements using ElementInputB = cutlass::int4b_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;