Bump torch-mlir to 5e4f00acb13f3f849a05e5ac28ee39307a5fdbff. (iree-or…

…g#17885) llvm/torch-mlir@5e4f00a --------- Signed-off-by: saienduri <saimanas.enduri@amd.com>
nod-ai · Jul 15, 2024 · a56975d · a56975d
1 parent c322d28
commit a56975d
Show file tree

Hide file tree

Showing 5 changed files with 1 addition and 25 deletions.
diff --git a/build_tools/pkgci/external_test_suite/onnx_cpu_llvm_sync.json b/build_tools/pkgci/external_test_suite/onnx_cpu_llvm_sync.json
@@ -377,11 +377,6 @@
     "onnx/node/generated/test_scatter_elements_with_duplicate_indices",
     "onnx/node/generated/test_scatter_elements_with_reduction_max",
     "onnx/node/generated/test_scatter_elements_with_reduction_min",
-    "onnx/node/generated/test_scatternd",
-    "onnx/node/generated/test_scatternd_add",
-    "onnx/node/generated/test_scatternd_max",
-    "onnx/node/generated/test_scatternd_min",
-    "onnx/node/generated/test_scatternd_multiply",
     "onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii",
     "onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_expanded",
     "onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_log_prob",
@@ -525,7 +520,6 @@
     "onnx/node/generated/test_castlike_FLOAT_to_DOUBLE_expanded",
     "onnx/node/generated/test_clip_default_int8_inbounds",
     "onnx/node/generated/test_clip_default_int8_inbounds_expanded",
-    "onnx/node/generated/test_constant_pad",
     "onnx/node/generated/test_constantofshape_float_ones",
     "onnx/node/generated/test_constantofshape_int_shape_zero",
     "onnx/node/generated/test_constantofshape_int_zeros",

diff --git a/build_tools/pkgci/external_test_suite/onnx_gpu_cuda.json b/build_tools/pkgci/external_test_suite/onnx_gpu_cuda.json
@@ -380,11 +380,6 @@
     "onnx/node/generated/test_scatter_elements_with_duplicate_indices",
     "onnx/node/generated/test_scatter_elements_with_reduction_max",
     "onnx/node/generated/test_scatter_elements_with_reduction_min",
-    "onnx/node/generated/test_scatternd",
-    "onnx/node/generated/test_scatternd_add",
-    "onnx/node/generated/test_scatternd_max",
-    "onnx/node/generated/test_scatternd_min",
-    "onnx/node/generated/test_scatternd_multiply",
     "onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii",
     "onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_expanded",
     "onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_log_prob",
@@ -528,7 +523,6 @@
     "onnx/node/generated/test_castlike_FLOAT_to_DOUBLE_expanded",
     "onnx/node/generated/test_clip_default_int8_inbounds",
     "onnx/node/generated/test_clip_default_int8_inbounds_expanded",
-    "onnx/node/generated/test_constant_pad",
     "onnx/node/generated/test_constantofshape_float_ones",
     "onnx/node/generated/test_constantofshape_int_shape_zero",
     "onnx/node/generated/test_constantofshape_int_zeros",

diff --git a/build_tools/pkgci/external_test_suite/onnx_gpu_rocm_rdna3.json b/build_tools/pkgci/external_test_suite/onnx_gpu_rocm_rdna3.json
@@ -375,11 +375,6 @@
     "onnx/node/generated/test_scatter_elements_with_duplicate_indices",
     "onnx/node/generated/test_scatter_elements_with_reduction_max",
     "onnx/node/generated/test_scatter_elements_with_reduction_min",
-    "onnx/node/generated/test_scatternd",
-    "onnx/node/generated/test_scatternd_add",
-    "onnx/node/generated/test_scatternd_max",
-    "onnx/node/generated/test_scatternd_min",
-    "onnx/node/generated/test_scatternd_multiply",
     "onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii",
     "onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_expanded",
     "onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_log_prob",
@@ -527,7 +522,6 @@
     "onnx/node/generated/test_castlike_FLOAT_to_FLOAT16_expanded",
     "onnx/node/generated/test_clip_default_int8_inbounds",
     "onnx/node/generated/test_clip_default_int8_inbounds_expanded",
-    "onnx/node/generated/test_constant_pad",
     "onnx/node/generated/test_constantofshape_float_ones",
     "onnx/node/generated/test_constantofshape_int_shape_zero",
     "onnx/node/generated/test_constantofshape_int_zeros",

diff --git a/build_tools/pkgci/external_test_suite/onnx_gpu_vulkan.json b/build_tools/pkgci/external_test_suite/onnx_gpu_vulkan.json
@@ -434,11 +434,6 @@
     "onnx/node/generated/test_scatter_elements_with_duplicate_indices",
     "onnx/node/generated/test_scatter_elements_with_reduction_max",
     "onnx/node/generated/test_scatter_elements_with_reduction_min",
-    "onnx/node/generated/test_scatternd",
-    "onnx/node/generated/test_scatternd_add",
-    "onnx/node/generated/test_scatternd_max",
-    "onnx/node/generated/test_scatternd_min",
-    "onnx/node/generated/test_scatternd_multiply",
     "onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii",
     "onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_expanded",
     "onnx/node/generated/test_sce_NCd1_mean_weight_negative_ii_log_prob",
@@ -600,7 +595,6 @@
     "onnx/node/generated/test_clip_default_int8_max_expanded",
     "onnx/node/generated/test_clip_default_int8_min",
     "onnx/node/generated/test_clip_default_int8_min_expanded",
-    "onnx/node/generated/test_constant_pad",
     "onnx/node/generated/test_constantofshape_float_ones",
     "onnx/node/generated/test_constantofshape_int_shape_zero",
     "onnx/node/generated/test_constantofshape_int_zeros",

diff --git a/third_party/torch-mlir b/third_party/torch-mlir
+1 −4		CMakeLists.txt
+7 −0		include/torch-mlir/Conversion/TorchOnnxToTorch/Utils.h
+14 −6		include/torch-mlir/Conversion/TorchToStablehlo/StablehloLegalizeUtils.h
+55 −0		include/torch-mlir/Dialect/Torch/IR/GeneratedTorchOps.td
+1 −1		lib/Conversion/TorchOnnxToTorch/CMakeLists.txt
+4 −3		lib/Conversion/TorchOnnxToTorch/DefaultDomainAtoF.cpp
+145 −84		lib/Conversion/TorchOnnxToTorch/DefaultDomainGtoP.cpp
+293 −0		lib/Conversion/TorchOnnxToTorch/DefaultDomainQtoZ.cpp
+0 −514		lib/Conversion/TorchOnnxToTorch/OnnxLstmExpander.cpp
+1,258 −0		lib/Conversion/TorchOnnxToTorch/OnnxRecurrentLayerOpExpanders.cpp
+12 −0		lib/Conversion/TorchOnnxToTorch/Utils.cpp
+64 −44		lib/Conversion/TorchToLinalg/TensorConstructors.cpp
+85 −75		lib/Conversion/TorchToStablehlo/Basic.cpp
+7 −16		lib/Conversion/TorchToStablehlo/GatherScatter.cpp
+14 −20		lib/Conversion/TorchToStablehlo/Linear.cpp
+8 −10		lib/Conversion/TorchToStablehlo/Pooling.cpp
+24 −42		lib/Conversion/TorchToStablehlo/Reduction.cpp
+52 −26		lib/Conversion/TorchToStablehlo/StablehloLegalizeUtils.cpp
+8 −10		lib/Conversion/TorchToStablehlo/ViewLike.cpp
+25 −6		lib/Conversion/TorchToTMTensor/TorchToTMTensor.cpp
+3 −1		lib/Dialect/Torch/IR/TorchOps.cpp
+57 −0		lib/Dialect/Torch/Transforms/AbstractInterpLibrary.cpp
+116 −29		lib/Dialect/Torch/Transforms/DecomposeComplexOps.cpp
+48 −3		lib/Dialect/Torch/Transforms/FuseQuantizedOps.cpp
+3 −14		projects/pt1/e2e_testing/xfail_sets.py
+31 −0		projects/pt1/python/torch_mlir/jit_ir_importer/build_tools/abstract_interp_lib_gen.py
+6 −0		projects/pt1/python/torch_mlir/jit_ir_importer/build_tools/torch_ods_gen.py
+27 −4		projects/pt1/python/torch_mlir_e2e_test/test_suite/elementwise.py
+25 −0		projects/pt1/python/torch_mlir_e2e_test/test_suite/scatter.py
+5 −1		python/CMakeLists.txt
+32 −0		test/Conversion/TorchOnnxToTorch/simple_ops_a_to_f.mlir
+87 −26		test/Conversion/TorchOnnxToTorch/simple_ops_g_to_p.mlir
+215 −0		test/Conversion/TorchOnnxToTorch/simple_ops_q_to_z.mlir
+24 −48		test/Conversion/TorchToStablehlo/linear.mlir
+7 −14		test/Conversion/TorchToStablehlo/pooling.mlir
+9 −11		test/Conversion/TorchToStablehlo/scatter.mlir
+15 −38		test/Conversion/TorchToStablehlo/view_like.mlir
+42 −0		test/Dialect/Torch/fuse-quantized-ops.mlir
+16 −1		test/Dialect/Torch/ops.mlir