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

fix description in example 12. (#444)

Browse Source 
Co-authored-by: Exusial <Exusial>
This commit is contained in:
Exusial 2022-04-25 04:29:06 +08:00 committed by GitHub
parent 4c0d6e1eb4
commit 310ed81ac3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
examples/12_gemm_bias_relu/gemm_bias_relu.cu
View File

@ -56,8 +56,10 @@ using ElementOutput = float; // <- data type of elements
// The code section below describes matrix layout of input and output matrices. // The code section below describes matrix layout of input and output matrices.
// Column Major for Matrix A, B and C. // Column Major for Matrix A, B and C.
//
// Note this example only works for ColumnMajor output because // Note that if the output is column major, the bias has to be per row. i.e. every row has different bias.
// If the output is row major, the bias has to be per column, i.e. every column has different bias.
// Below list some other notices:
// 1) we only have row major epilogue. // 1) we only have row major epilogue.
// 2) we swap A and B if the output is column major then we can still use the // 2) we swap A and B if the output is column major then we can still use the
// row major epilogue. // row major epilogue.