squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Modify comments in code examples/08_turing_tensorop_gemm/turing_tensorop_gemm.cu (#1325)

Browse Source
This commit is contained in:
xws117 2024-02-01 10:41:30 +08:00 committed by GitHub
parent 8825fbf1ef
commit 6e3df975a2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/08_turing_tensorop_gemm/turing_tensorop_gemm.cu
View File

@ -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 ElementInputB = int8_t; // <- data type of elements in input matrix B
using ElementOutput = int32_t; // <- data type of elements in output matrix D 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 // The code section below describes matrix layout of input and output matrices. Row Major for
// Matrix A, Row Major for Matrix B and Row Major for Matrix C // Matrix A, Column Major for Matrix B and Row Major for Matrix C
using LayoutInputA = cutlass::layout::RowMajor; using LayoutInputA = cutlass::layout::RowMajor;
using LayoutInputB = cutlass::layout::ColumnMajor; using LayoutInputB = cutlass::layout::ColumnMajor;
using LayoutOutput = cutlass::layout::RowMajor; using LayoutOutput = cutlass::layout::RowMajor;