From b6ccf34aef08a56195ce8465807d35a51e01acae Mon Sep 17 00:00:00 2001
From: Lifu Huang <lifu.hlf@gmail.com>
Date: Thu, 4 Sep 2025 16:56:17 -0700
Subject: [PATCH] Fix Copy_Atom type mismatch in sgemm_sm80.cu (#2582)

---
 examples/cute/tutorial/sgemm_sm80.cu | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/examples/cute/tutorial/sgemm_sm80.cu b/examples/cute/tutorial/sgemm_sm80.cu
index 29fb51e5..bd7c6fa4 100644
--- a/examples/cute/tutorial/sgemm_sm80.cu
+++ b/examples/cute/tutorial/sgemm_sm80.cu
@@ -496,8 +496,8 @@ gemm_nt(int m, int n, int k,
                size(ceil_div(N, bN)));
   gemm_device<<<dimGrid, dimBlock, smem_size, stream>>>
       (prob_shape, cta_tiler,
-       A, dA, sA, copyA, AutoVectorizingCopy{},
-       B, dB, sB, copyB, AutoVectorizingCopy{},
+       A, dA, sA, copyA, Copy_Atom<AutoVectorizingCopy, TA>{},
+       B, dB, sB, copyB, Copy_Atom<AutoVectorizingCopy, TB>{},
        C, dC, sC, mmaC,
        alpha, beta);
 }
@@ -573,8 +573,8 @@ gemm_tn(int m, int n, int k,
                size(ceil_div(N, bN)));
   gemm_device<<<dimGrid, dimBlock, smem_size, stream>>>
       (prob_shape, cta_tiler,
-       A, dA, sA, copyA, AutoVectorizingCopy{},
-       B, dB, sB, copyB, AutoVectorizingCopy{},
+       A, dA, sA, copyA, Copy_Atom<AutoVectorizingCopy, TA>{},
+       B, dB, sB, copyB, Copy_Atom<AutoVectorizingCopy, TB>{},
        C, dC, sC, mmaC,
        alpha, beta);
 }