Skip to content

Commit

Permalink
Modify comments in code examples/08_turing_tensorop_gemm/turing_tenso…
Browse files Browse the repository at this point in the history
…rop_gemm.cu (NVIDIA#1325)
  • Loading branch information
xws117 authored Feb 1, 2024
1 parent 8825fbf commit 6e3df97
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions examples/08_turing_tensorop_gemm/turing_tensorop_gemm.cu
Original file line number Diff line number Diff line change
Expand Up @@ -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;
Expand Down

0 comments on commit 6e3df97

Please sign in to comment.