From 310ed81ac3f3bd42d1f81fffd183addb0aa931d1 Mon Sep 17 00:00:00 2001 From: Exusial <51511617+Exusial@users.noreply.github.com> Date: Mon, 25 Apr 2022 04:29:06 +0800 Subject: [PATCH] fix description in example 12. (#444) Co-authored-by: Exusial --- examples/12_gemm_bias_relu/gemm_bias_relu.cu | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/examples/12_gemm_bias_relu/gemm_bias_relu.cu b/examples/12_gemm_bias_relu/gemm_bias_relu.cu index 43808354..62eb2940 100644 --- a/examples/12_gemm_bias_relu/gemm_bias_relu.cu +++ b/examples/12_gemm_bias_relu/gemm_bias_relu.cu @@ -56,8 +56,10 @@ using ElementOutput = float; // <- data type of elements // The code section below describes matrix layout of input and output matrices. // 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. // 2) we swap A and B if the output is column major then we can still use the // row major epilogue.