correct BLayout stride in SM80 m16n8k32 int4 MMA traits (#3140)

2026-04-21 17:17:03 +08:00
parent ea46e277d2
commit aeba0d3723
1 changed files with 1 additions and 1 deletions
--- 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;
 };