adding bias

83511d7e · Chao Liu · 30109f67 · 83511d7e
Commit 83511d7e authored Jun 01, 2022 by Chao Liu
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 3 deletions

include/ck/tensor_operation/gpu/device/device_gemm_bias_activation_xdl_cshuffle.hpp ...n/gpu/device/device_gemm_bias_activation_xdl_cshuffle.hpp +5 -3

No files found.
--- a/include/ck/tensor_operation/gpu/device/device_gemm_bias_activation_xdl_cshuffle.hpp
+++ b/include/ck/tensor_operation/gpu/device/device_gemm_bias_activation_xdl_cshuffle.hpp
@@ -76,9 +76,11 @@ namespace ck {
 namespace tensor_operation {
 namespace device {
-// Note: inter-wave loop scheduler is rolled out to c-shuffle version first. Becuase non c-shuffle
+// C = c_element_op(a_element_op(A) * b_element_op(B), D))
-// version currently has compiler issues with register spill which further causes validation
+// A: [M, K] or [K, M]
-// failures.
+// B: [N, K] or [K, N]
+// C: [M, N]
+// D: [N]
 template <typename ALayout,
          typename BLayout,
          typename CLayout,