try fix ldmatrix b for int8

masahi · masahi · commit 6fa91e55b5ab · 2022-05-17T10:28:34.000+09:00
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",
             )
         )

Original file line number	Diff line number	Diff line change
`@@ -106,7 +106,7 @@ def ldmatrix_b_impl(a: T.handle, c: T.handle) -> None:`
`106`	`106`	`B_warp.data,`
`107`	`107`	`16 * tx,`
`108`	`108`	`B_shared.data,`
`109`		`- 16 * tx,`
	`109`	`+ 32 * (tx % 16) + 8 * (tx // 16),`
`110`	`110`	`dtype="int8",`
`111`	`111`	`)`
`112`	`112`	`)`