This commit is contained in:
reed
2023-08-27 00:49:26 -04:00
committed by GitHub
parent 7618e9bfd8
commit 6673df0e48
+1 -1
View File
@@ -40,7 +40,7 @@
#include <cute/atom/mma_atom.hpp> #include <cute/atom/mma_atom.hpp>
/** The gemm algorithm takes four (or three) tensors and computes /** The gemm algorithm takes four (or three) tensors and computes
* D += A * B + C * D = A * B + C
* It dispatches based on the number of modes each tensor has: * It dispatches based on the number of modes each tensor has:
* *
* 1. `(V) x (V) => (V)`. * 1. `(V) x (V) => (V)`.