Skip to content

Commit

Permalink
Bump torch-mlir to 7e7af670802d99cacdaf26e6e37249d544e4896e. (iree-or…
Browse files Browse the repository at this point in the history 
…g#17885) (iree-org#18097)

llvm/torch-mlir@7e7af67

---------

Signed-off-by: Rob Suderman <[email protected]>
  • Loading branch information
@rsuderman
rsuderman authored Aug 5, 2024
1 parent 345e655 commit 0f3bea2
Show file tree
Hide file tree
Showing 5 changed files with 2 additions and 26 deletions.
6 changes: 0 additions & 6 deletions build_tools/pkgci/external_test_suite/onnx_cpu_llvm_sync.json
View file
Original file line number Diff line number Diff line change
Expand Up @@ -521,7 +521,6 @@
"onnx/node/generated/test_constantofshape_int_zeros",
"onnx/node/generated/test_dropout_default_mask_ratio",
"onnx/node/generated/test_einsum_transpose",
"onnx/node/generated/test_elu_default",
"onnx/node/generated/test_eyelike_with_dtype",
"onnx/node/generated/test_gather_elements_negative_indices",
"onnx/node/generated/test_gridsample_nearest",
Expand Down Expand Up @@ -551,11 +550,6 @@
"onnx/node/generated/test_reduce_min_empty_set",
"onnx/node/generated/test_reduce_sum_empty_set_non_reduced_axis_zero",
"onnx/node/generated/test_resize_downsample_scales_linear_align_corners",
"onnx/node/generated/test_scatter_elements_with_negative_indices",
"onnx/node/generated/test_sce_mean_no_weight_ii",
"onnx/node/generated/test_sce_mean_no_weight_ii_expanded",
"onnx/node/generated/test_sce_mean_no_weight_ii_log_prob",
"onnx/node/generated/test_sce_mean_no_weight_ii_log_prob_expanded",
"onnx/node/generated/test_shape_end_1",
"onnx/node/generated/test_shape_start_1",
"onnx/node/generated/test_shape_start_1_end_2",
Expand Down
7 changes: 1 addition & 6 deletions build_tools/pkgci/external_test_suite/onnx_gpu_cuda.json
View file
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
"skip_compile_tests": [
"onnx/node/generated/test_dequantizelinear",
"onnx/node/generated/test_group_normalization_example_expanded",
"onnx/node/generated/test_group_normalization_epsilon_expanded",
"onnx/node/generated/test_group_normalization_epsilon_expanded"
],
"skip_run_tests": [
"onnx/node/generated/test_gather_elements_negative_indices",
Expand Down Expand Up @@ -524,7 +524,6 @@
"onnx/node/generated/test_constantofshape_int_zeros",
"onnx/node/generated/test_dropout_default_mask_ratio",
"onnx/node/generated/test_einsum_transpose",
"onnx/node/generated/test_elu_default",
"onnx/node/generated/test_eyelike_with_dtype",
"onnx/node/generated/test_gridsample_nearest",
"onnx/node/generated/test_gridsample_nearest_align_corners_0_additional_1",
Expand Down Expand Up @@ -569,10 +568,6 @@
"onnx/node/generated/test_reduce_sum_square_default_axes_keepdims_example_expanded",
"onnx/node/generated/test_reduce_sum_square_default_axes_keepdims_random",
"onnx/node/generated/test_reduce_sum_square_default_axes_keepdims_random_expanded",
"onnx/node/generated/test_sce_mean_no_weight_ii",
"onnx/node/generated/test_sce_mean_no_weight_ii_expanded",
"onnx/node/generated/test_sce_mean_no_weight_ii_log_prob",
"onnx/node/generated/test_sce_mean_no_weight_ii_log_prob_expanded",
"onnx/node/generated/test_shape_end_1",
"onnx/node/generated/test_shape_start_1",
"onnx/node/generated/test_shape_start_1_end_2",
Expand Down
7 changes: 0 additions & 7 deletions build_tools/pkgci/external_test_suite/onnx_gpu_rocm_rdna3.json
View file
Original file line number Diff line number Diff line change
Expand Up @@ -523,7 +523,6 @@
"onnx/node/generated/test_constantofshape_int_zeros",
"onnx/node/generated/test_dropout_default_mask_ratio",
"onnx/node/generated/test_einsum_transpose",
"onnx/node/generated/test_elu_default",
"onnx/node/generated/test_expand_dim_changed",
"onnx/node/generated/test_expand_dim_unchanged",
"onnx/node/generated/test_eyelike_populate_off_main_diagonal",
Expand Down Expand Up @@ -580,12 +579,6 @@
"onnx/node/generated/test_reduce_sum_square_default_axes_keepdims_random",
"onnx/node/generated/test_reduce_sum_square_default_axes_keepdims_random_expanded",
"onnx/node/generated/test_resize_downsample_scales_linear_align_corners",
"onnx/node/generated/test_resize_downsample_scales_linear_half_pixel_symmetric",
"onnx/node/generated/test_scatter_elements_with_negative_indices",
"onnx/node/generated/test_sce_mean_no_weight_ii",
"onnx/node/generated/test_sce_mean_no_weight_ii_expanded",
"onnx/node/generated/test_sce_mean_no_weight_ii_log_prob",
"onnx/node/generated/test_sce_mean_no_weight_ii_log_prob_expanded",
"onnx/node/generated/test_shape",
"onnx/node/generated/test_shape_clip_end",
"onnx/node/generated/test_shape_clip_start",
Expand Down
6 changes: 0 additions & 6 deletions build_tools/pkgci/external_test_suite/onnx_gpu_vulkan.json
View file
Original file line number Diff line number Diff line change
Expand Up @@ -604,7 +604,6 @@
"onnx/node/generated/test_dynamicquantizelinear",
"onnx/node/generated/test_dynamicquantizelinear_expanded",
"onnx/node/generated/test_einsum_transpose",
"onnx/node/generated/test_elu_default",
"onnx/node/generated/test_eyelike_with_dtype",
"onnx/node/generated/test_gather_elements_negative_indices",
"onnx/node/generated/test_isinf_float16",
Expand Down Expand Up @@ -664,11 +663,6 @@
"onnx/node/generated/test_reduce_sum_square_default_axes_keepdims_example_expanded",
"onnx/node/generated/test_reduce_sum_square_default_axes_keepdims_random",
"onnx/node/generated/test_reduce_sum_square_default_axes_keepdims_random_expanded",
"onnx/node/generated/test_scatter_elements_with_negative_indices",
"onnx/node/generated/test_sce_mean_no_weight_ii",
"onnx/node/generated/test_sce_mean_no_weight_ii_expanded",
"onnx/node/generated/test_sce_mean_no_weight_ii_log_prob",
"onnx/node/generated/test_sce_mean_no_weight_ii_log_prob_expanded",
"onnx/node/generated/test_shape_end_1",
"onnx/node/generated/test_shape_start_1",
"onnx/node/generated/test_shape_start_1_end_2",
Expand Down
2 changes: 1 addition & 1 deletion third_party/torch-mlir
Submodule torch-mlir updated 58 files
+4 −4 .github/workflows/RollPyTorch.yml
+2 −2 build_tools/python_deploy/build_linux_packages.sh
+1 −1 build_tools/python_deploy/build_windows.ps1
+1 −1 externals/stablehlo
+10 −0 include/torch-mlir/Conversion/TorchOnnxToTorch/Patterns.h
+6 −1 include/torch-mlir/Conversion/TorchToStablehlo/StablehloLegalizeUtils.h
+269 −0 include/torch-mlir/Dialect/Torch/IR/GeneratedTorchOps.td
+2 −2 lib/Conversion/TorchOnnxToTorch/DefaultDomainAtoF.cpp
+207 −0 lib/Conversion/TorchOnnxToTorch/DefaultDomainGtoP.cpp
+305 −2 lib/Conversion/TorchOnnxToTorch/DefaultDomainQtoZ.cpp
+61 −25 lib/Conversion/TorchToLinalg/Linear.cpp
+257 −0 lib/Conversion/TorchToLinalg/Pooling.cpp
+329 −16 lib/Conversion/TorchToLinalg/Random.cpp
+40 −20 lib/Conversion/TorchToLinalg/Uncategorized.cpp
+12 −7 lib/Conversion/TorchToStablehlo/Basic.cpp
+66 −33 lib/Conversion/TorchToStablehlo/GatherScatter.cpp
+84 −1 lib/Conversion/TorchToStablehlo/StablehloLegalizeUtils.cpp
+118 −5 lib/Conversion/TorchToTosa/TorchToTosa.cpp
+182 −134 lib/Dialect/Torch/IR/TorchOps.cpp
+454 −10 lib/Dialect/Torch/Transforms/AbstractInterpLibrary.cpp
+573 −28 lib/Dialect/Torch/Transforms/DecomposeComplexOps.cpp
+6 −0 lib/Dialect/Torch/Transforms/LowerToBackendContract.cpp
+26 −0 lib/Dialect/TorchConversion/Transforms/BackendTypeConversionPasses.cpp
+7 −0 lib/Dialect/TorchConversion/Transforms/Passes.cpp
+2 −0 lib/InitAll.cpp
+57 −21 projects/pt1/e2e_testing/xfail_sets.py
+150 −10 projects/pt1/python/torch_mlir/jit_ir_importer/build_tools/abstract_interp_lib_gen.py
+26 −1 projects/pt1/python/torch_mlir/jit_ir_importer/build_tools/torch_ods_gen.py
+1 −0 projects/pt1/python/torch_mlir_e2e_test/test_suite/__init__.py
+24 −0 projects/pt1/python/torch_mlir_e2e_test/test_suite/basic.py
+64 −13 projects/pt1/python/torch_mlir_e2e_test/test_suite/conv.py
+79 −0 projects/pt1/python/torch_mlir_e2e_test/test_suite/elementwise.py
+42 −0 projects/pt1/python/torch_mlir_e2e_test/test_suite/linalg_algorithms.py
+25 −0 projects/pt1/python/torch_mlir_e2e_test/test_suite/norm_like.py
+81 −0 projects/pt1/python/torch_mlir_e2e_test/test_suite/pooling.py
+43 −0 projects/pt1/python/torch_mlir_e2e_test/test_suite/reshape_like.py
+47 −0 projects/pt1/python/torch_mlir_e2e_test/test_suite/rng.py
+18 −0 projects/pt1/python/torch_mlir_e2e_test/test_suite/scalar.py
+53 −0 projects/pt1/python/torch_mlir_e2e_test/test_suite/spectral.py
+8 −7 python/torch_mlir/compiler_utils.py
+4 −1 python/torch_mlir/extras/fx_decomp_util.py
+64 −23 python/torch_mlir/extras/fx_importer.py
+9 −4 python/torch_mlir/fx.py
+1 −1 pytorch-hash.txt
+2 −2 pytorch-requirements.txt
+1 −1 test/Conversion/TorchOnnxToTorch/simple_ops_a_to_f.mlir
+135 −0 test/Conversion/TorchOnnxToTorch/simple_ops_g_to_p.mlir
+336 −8 test/Conversion/TorchOnnxToTorch/simple_ops_q_to_z.mlir
+31 −4 test/Conversion/TorchToLinalg/resize.mlir
+18 −8 test/Conversion/TorchToStablehlo/view_like.mlir
+89 −16 test/Dialect/Torch/canonicalize.mlir
+34 −31 test/Dialect/Torch/decompose-complex-ops.mlir
+10 −0 test/Dialect/TorchConversion/finalizing-backend-type-conversion.mlir
+2 −2 test/python/fx_importer/basic_test.py
+48 −1 test/python/fx_importer/custom_op_test.py
+51 −16 test/python/fx_importer/symbolic_shape_expr_test.py
+1 −1 test/python/fx_importer/v2.3/types_test.py
+2 −2 torchvision-requirements.txt

0 comments on commit 0f3bea2

Please sign in to comment.