[all-commits] [llvm/llvm-project] 9420fc: [MLIR] SM_90 integratation test of TMA `128x64xf16...

Guray Ozen via All-commits all-commits at lists.llvm.org
Fri Sep 15 00:30:30 PDT 2023


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 9420fc4956d0d6b37d87c52dcbadeb2416b5488a
      https://github.com/llvm/llvm-project/commit/9420fc4956d0d6b37d87c52dcbadeb2416b5488a
  Author: Guray Ozen <guray.ozen at gmail.com>
  Date:   2023-09-15 (Fri, 15 Sep 2023)

  Changed paths:
    A mlir/test/Integration/GPU/CUDA/sm90/tma_load_64x64_swizzle128b.mlir

  Log Message:
  -----------
  [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.




More information about the All-commits mailing list