[torch-frontend] change setupBackendTypeConversion to setupBackendTyp… (

#360) …eConversionForStablehlo
bytedance · Jun 20, 2024 · f187897 · f187897
1 parent e5248af
commit f187897
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/frontends/torch-frontend/torch-frontend/lib/Conversion/ConvertTorchToCcl.cpp b/frontends/torch-frontend/torch-frontend/lib/Conversion/ConvertTorchToCcl.cpp
@@ -299,7 +299,8 @@ class ConvertTorchToCcl : public ConvertTorchToCclBase<ConvertTorchToCcl> {
 
     TypeConverter typeConverter;
     typeConverter.addConversion([](Type type) { return type; });
-    TorchConversion::setupBackendTypeConversion(target, typeConverter);
+    TorchConversion::setupBackendTypeConversionForStablehlo(target,
+                                                            typeConverter);
 
     RewritePatternSet patterns(context);
 

diff --git a/frontends/torch-frontend/torch-frontend/lib/Conversion/ConvertTorchToCustomCall.cpp b/frontends/torch-frontend/torch-frontend/lib/Conversion/ConvertTorchToCustomCall.cpp
@@ -1232,7 +1232,8 @@ class ConvertTorchToCustomCall
 
     TypeConverter typeConverter;
     typeConverter.addConversion([](Type type) { return type; });
-    TorchConversion::setupBackendTypeConversion(target, typeConverter);
+    TorchConversion::setupBackendTypeConversionForStablehlo(target,
+                                                            typeConverter);
 
     RewritePatternSet patterns(context);
     if (validCustomCallOpsSet.contains("aten.native_layer_norm")) {

diff --git a/frontends/torch-frontend/torch-frontend/lib/Conversion/ConvertTorchToStablehloExt.cpp b/frontends/torch-frontend/torch-frontend/lib/Conversion/ConvertTorchToStablehloExt.cpp
@@ -345,7 +345,8 @@ struct ConvertTorchToStablehloExtPass
                            arith::ArithDialect>();
     TypeConverter typeConverter;
     typeConverter.addConversion([](Type type) { return type; });
-    TorchConversion::setupBackendTypeConversion(target, typeConverter);
+    TorchConversion::setupBackendTypeConversionForStablehlo(target,
+                                                            typeConverter);
 
     RewritePatternSet patterns(context);
     target.addIllegalOp<Aten_IndexPutImplOp>();