From aeba0d3723b8c226b7180a59081ae1cdfdf087a8 Mon Sep 17 00:00:00 2001
From: zfm <109492464+zfmmmm@users.noreply.github.com>
Date: Tue, 21 Apr 2026 17:17:03 +0800
Subject: [PATCH] correct BLayout stride in SM80 m16n8k32 int4 MMA traits
 (#3140)

---
 include/cute/atom/mma_traits_sm80.hpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/include/cute/atom/mma_traits_sm80.hpp b/include/cute/atom/mma_traits_sm80.hpp
index 357921e4..61eea7c2 100644
--- a/include/cute/atom/mma_traits_sm80.hpp
+++ b/include/cute/atom/mma_traits_sm80.hpp
@@ -458,7 +458,7 @@ struct MMA_Traits<SM80_16x8x32_S32S4S4S32_TN> {
                          Stride<Stride<_128, _1>, Stride<_16, _8>>>;
   // (T32,V8) -> (M8,N32)
   using BLayout = Layout<Shape <Shape < _4, _8>, Shape <_8>>,
-                         Stride<Stride<_32, _1>, Stride<_8>>>;
+                         Stride<Stride<_64, _1>, Stride<_8>>>;
   using CLayout = SM80_16x8_Row;
 };