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

Update gemm_api.md

Browse Source
This commit is contained in:
Haicheng Wu 2023-05-12 15:37:31 -04:00 committed by GitHub
parent 19c4a4815e
commit e2953d47c5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
media/docs/gemm_api.md
View File

@ -28,7 +28,7 @@ for (int cta_n = 0; cta_n < GemmN; cta_n += CtaTileN) { // f
for (int warp_n = 0; warp_n < CtaTileN; warp_n += WarpTileN) { // for each warp } warp-level concurrency for (int warp_n = 0; warp_n < CtaTileN; warp_n += WarpTileN) { // for each warp } warp-level concurrency
for (int warp_m = 0; warp_m < CtaTileM; warp_m += WarpTileM) { // for each warp } for (int warp_m = 0; warp_m < CtaTileM; warp_m += WarpTileM) { // for each warp }
// //
for (int warp_k = 0; warp_k < CtaTileK; warp_k += MmaK) { // fully unroll across CtaTileK - one iteration of this loop is one "k Group" == "MmaK" for (int warp_k = 0; warp_k < CtaTileK; warp_k += WarpTileK) { // fully unroll across CtaTileK - one iteration of this loop is one "k Group"
// //
for (int mma_k = 0; mma_k < WarpTileK; mma_k += MmaK) { // cutlass::gemm::warp::Mma for (int mma_k = 0; mma_k < WarpTileK; mma_k += MmaK) { // cutlass::gemm::warp::Mma
for (int mma_n = 0; mma_n < WarpTileN; mma_n += MmaN) { // for (int mma_n = 0; mma_n < WarpTileN; mma_n += MmaN) { //