[Triton-MLIR][Backend] Support layout conversion between mmaLayout and blockedLayout (#693)

2022-09-27 11:58:47 +08:00
parent 1e91ed30d0
commit 61b61755e5
6 changed files with 205 additions and 104 deletions
--- a/include/triton/Dialect/TritonGPU/IR/Dialect.h
+++ b/include/triton/Dialect/TritonGPU/IR/Dialect.h
@@ -22,8 +22,12 @@ namespace gpu {

 unsigned getElemsPerThread(Attribute layout, ArrayRef<int64_t> shape);

+SmallVector<unsigned> getSizePerThread(Attribute layout);
+
 unsigned getShapePerCTA(const Attribute &layout, unsigned d);

+SmallVector<unsigned> getOrder(const Attribute &layout);
+
 } // namespace gpu
 } // namespace triton
 } // namespace mlir