Switch to torch decomposition from linalg lowering

llvm · Nov 13, 2024 · fe96416 · fe96416
1 parent 5a4f5fc
commit fe96416
Show file tree

Hide file tree

Showing 5 changed files with 280 additions and 185 deletions.
diff --git a/lib/Conversion/TorchToLinalg/CMakeLists.txt b/lib/Conversion/TorchToLinalg/CMakeLists.txt
@@ -8,7 +8,6 @@ add_mlir_conversion_library(TorchMLIRTorchToLinalg
   TensorConstructors.cpp
   TensorScalarInterop.cpp
   TorchToLinalg.cpp
-  TorchvisionOp.cpp
   Uncategorized.cpp
   Utils.cpp
 

diff --git a/lib/Conversion/TorchToLinalg/PopulatePatterns.h b/lib/Conversion/TorchToLinalg/PopulatePatterns.h
@@ -63,9 +63,6 @@ void populateIndirectDataMovementPatternsAndLegality(
 void populateTensorConstructorsPatternsAndLegality(TypeConverter &typeConverter,
                                                    RewritePatternSet &patterns,
                                                    ConversionTarget &target);
-void populateTorchvisionPatternsAndLegality(TypeConverter &typeConverter,
-                                            RewritePatternSet &patterns,
-                                            ConversionTarget &target);
 
 } // namespace torch_to_linalg
 } // namespace torch

diff --git a/lib/Conversion/TorchToLinalg/TorchToLinalg.cpp b/lib/Conversion/TorchToLinalg/TorchToLinalg.cpp
@@ -81,8 +81,6 @@ class ConvertTorchToLinalg
         typeConverter, patterns, target);
     torch_to_linalg::populateTensorConstructorsPatternsAndLegality(
         typeConverter, patterns, target);
-    torch_to_linalg::populateTorchvisionPatternsAndLegality(typeConverter,
-                                                            patterns, target);
 
     if (failed(applyPartialConversion(getOperation(), target,
                                       std::move(patterns))))

diff --git a/lib/Conversion/TorchToLinalg/TorchvisionOp.cpp b/lib/Conversion/TorchToLinalg/TorchvisionOp.cpp