try fix ldmatrix b for int8

masahi · May 17, 2022 · 6fa91e5 · 6fa91e5
1 parent 7a962cd
commit 6fa91e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/unittest/test_mma_16x8x32_int8.py b/tests/python/unittest/test_mma_16x8x32_int8.py
@@ -106,7 +106,7 @@ def ldmatrix_b_impl(a: T.handle, c: T.handle) -> None:
                 B_warp.data,
                 16 * tx,
                 B_shared.data,
-                16 * tx,
+                32 * (tx % 16) + 8 * (tx // 16),
                 dtype="int8",
             )
         )