[Analysis] Added Axis Info Analysis (#8)

2022-07-19 13:38:48 -07:00
parent df940aaab0
commit a633d2b403
20 changed files with 582 additions and 13 deletions
--- a/lib/Conversion/TritonToTritonGPU/TritonToTritonGPU.cpp
+++ b/lib/Conversion/TritonToTritonGPU/TritonToTritonGPU.cpp
@@ -240,6 +240,7 @@ void populateTritonPatterns(
 ) {
  MLIRContext *context = patterns.getContext();
  patterns.add<TritonGenericPattern<triton::ReshapeOp>,
+               TritonGenericPattern<triton::SplatOp>,
               TritonGenericPattern<triton::BroadcastOp>,
               TritonGenericPattern<triton::GEPOp>,
               TritonReducePattern,