[BACKEND] Add generic support of convert_layout from distributed to shared (#1025)

2022-12-31 03:29:58 +08:00
parent 194ba103b1
commit 0e8590f1c9
5 changed files with 68 additions and 110 deletions
--- a/include/triton/Dialect/TritonGPU/IR/Dialect.h
+++ b/include/triton/Dialect/TritonGPU/IR/Dialect.h
@@ -39,6 +39,8 @@ SmallVector<unsigned> getShapePerCTA(const Attribute &layout);

 SmallVector<unsigned> getOrder(const Attribute &layout);

+bool isaDistributedLayout(const Attribute &layout);
+
 } // namespace gpu
 } // namespace triton
 } // namespace mlir