[MHLO] Init MHLO reduce-like op conversion (#1133)

* [MHLO] init reduce-like op conversion from Torch to MHLO Co-authored-by: Bairen Yi <[email protected]> Co-authored-by: Jiawei Wu <[email protected]> Co-authored-by: Tianyou Guo <[email protected]> Co-authored-by: Xu Yan <[email protected]> Co-authored-by: Ziheng Jiang <[email protected]>
llvm · Aug 3, 2022 · 636f5ac · 636f5ac
1 parent 0b23af2
commit 636f5ac
Show file tree

Hide file tree

Showing 5 changed files with 815 additions and 1 deletion.
diff --git a/lib/Conversion/TorchToMhlo/CMakeLists.txt b/lib/Conversion/TorchToMhlo/CMakeLists.txt
@@ -4,6 +4,7 @@ add_mlir_conversion_library(TorchMLIRTorchToMhlo
   BasicOp.cpp
   GatherOp.cpp
   ViewLikeOps.cpp
+  ReductionOp.cpp
 
   ADDITIONAL_HEADER_DIRS
   ${PROJECT_SOURCE_DIR}/include/torch-mlir/Conversion/TorchToMhlo

diff --git a/lib/Conversion/TorchToMhlo/PopulatePatterns.h b/lib/Conversion/TorchToMhlo/PopulatePatterns.h
@@ -25,7 +25,9 @@ void populateViewLikeOpPatternsAndLegality(TypeConverter &typeConverter,
 void populateGatherOpPatternsAndLegality(TypeConverter &typeConverter,
                                          RewritePatternSet &patterns,
                                          ConversionTarget &target);
-
+void populateReductionOpPatternsAndLegality(TypeConverter &typeConverter,
+                                        RewritePatternSet &patterns,
+                                        ConversionTarget &target);
 } // namespace torch_to_mhlo
 } // namespace torch
 } // namespace mlir