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

Fix Copy_Atom type mismatch in sgemm_sm80.cu (#2582)

Browse Source
This commit is contained in:
Lifu Huang
2025-09-04 16:56:17 -07:00
committed by GitHub
parent 2288c0c901
commit b6ccf34aef
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
examples/cute/tutorial/sgemm_sm80.cu
View File

@ -496,8 +496,8 @@ gemm_nt(int m, int n, int k,
size(ceil_div(N, bN))); size(ceil_div(N, bN)));
gemm_device<<<dimGrid, dimBlock, smem_size, stream>>> gemm_device<<<dimGrid, dimBlock, smem_size, stream>>>
(prob_shape, cta_tiler, (prob_shape, cta_tiler,
A, dA, sA, copyA, AutoVectorizingCopy{}, A, dA, sA, copyA, Copy_Atom<AutoVectorizingCopy, TA>{},
B, dB, sB, copyB, AutoVectorizingCopy{}, B, dB, sB, copyB, Copy_Atom<AutoVectorizingCopy, TB>{},
C, dC, sC, mmaC, C, dC, sC, mmaC,
alpha, beta); alpha, beta);
} }
@ -573,8 +573,8 @@ gemm_tn(int m, int n, int k,
size(ceil_div(N, bN))); size(ceil_div(N, bN)));
gemm_device<<<dimGrid, dimBlock, smem_size, stream>>> gemm_device<<<dimGrid, dimBlock, smem_size, stream>>>
(prob_shape, cta_tiler, (prob_shape, cta_tiler,
A, dA, sA, copyA, AutoVectorizingCopy{}, A, dA, sA, copyA, Copy_Atom<AutoVectorizingCopy, TA>{},
B, dB, sB, copyB, AutoVectorizingCopy{}, B, dB, sB, copyB, Copy_Atom<AutoVectorizingCopy, TB>{},
C, dC, sC, mmaC, C, dC, sC, mmaC,
alpha, beta); alpha, beta);
} }