Skip to content

Commit 9420fc4

Browse files
authored
[MLIR] SM_90 integratation test of TMA 128x64xf16 and 64x64xf16 with 128b Swizzling (#65954)
The #65953 added a test `128x64xf16` that does a single TMA load. This PR adds more complex test that does 2 additional TMA loads with 128B Swizzling: ``` TMA Load: Matrix-A[0:128][0:64] TMA Load: Matrix-B[0:64][0:64] TMA Load: Matrix-B[64:128][0:64] ``` The program tests the loaded data for Matrix-B.
1 parent 07460b6 commit 9420fc4

File tree

1 file changed

+247
-0
lines changed

1 file changed

+247
-0
lines changed

0 commit comments

Comments
 (0)