Update torch-mlir to 1af00e6 (iree-org#17286)

nithinsubbiah · May 7, 2024 · d8f49dc · d8f49dc
1 parent 6e76f28
commit d8f49dc
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 28 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
@@ -58,8 +58,6 @@
     "test_bernoulli_double",
     "test_bernoulli_seed",
     "test_bernoulli_seed_expanded",
-    "test_blackmanwindow",
-    "test_blackmanwindow_symmetric",
     "test_cast_FLOAT16_to_FLOAT8E4M3FN",
     "test_cast_FLOAT16_to_FLOAT8E4M3FNUZ",
     "test_cast_FLOAT16_to_FLOAT8E5M2",
@@ -191,8 +189,6 @@
     "test_gru_with_initial_bias",
     "test_hammingwindow",
     "test_hammingwindow_symmetric",
-    "test_hannwindow",
-    "test_hannwindow_symmetric",
     "test_hardmax_axis_0",
     "test_hardmax_axis_1",
     "test_hardmax_axis_2",
@@ -380,8 +376,6 @@
     "test_reduce_sum_keepdims_example",
     "test_reduce_sum_keepdims_random",
     "test_reduce_sum_negative_axes_keepdims_example",
-    "test_reduce_sum_square_default_axes_keepdims_example",
-    "test_reduce_sum_square_default_axes_keepdims_random",
     "test_reduce_sum_square_do_not_keepdims_example",
     "test_reduce_sum_square_do_not_keepdims_example_expanded",
     "test_reduce_sum_square_do_not_keepdims_random",
@@ -653,7 +647,6 @@
     "test_pow_types_float32_int64",
     "test_pow_types_float32_uint32",
     "test_pow_types_float32_uint64",
-    "test_qlinearconv",
     "test_qlinearmatmul_2D_int8_float16",
     "test_qlinearmatmul_2D_int8_float32",
     "test_qlinearmatmul_3D_int8_float16",

diff --git a/build_tools/pkgci/external_test_suite/onnx_gpu_cuda.json b/build_tools/pkgci/external_test_suite/onnx_gpu_cuda.json
@@ -62,8 +62,6 @@
     "test_bernoulli_double",
     "test_bernoulli_seed",
     "test_bernoulli_seed_expanded",
-    "test_blackmanwindow",
-    "test_blackmanwindow_symmetric",
     "test_cast_FLOAT16_to_FLOAT8E4M3FN",
     "test_cast_FLOAT16_to_FLOAT8E4M3FNUZ",
     "test_cast_FLOAT16_to_FLOAT8E5M2",
@@ -195,8 +193,6 @@
     "test_gru_with_initial_bias",
     "test_hammingwindow",
     "test_hammingwindow_symmetric",
-    "test_hannwindow",
-    "test_hannwindow_symmetric",
     "test_hardmax_axis_0",
     "test_hardmax_axis_1",
     "test_hardmax_axis_2",
@@ -381,8 +377,6 @@
     "test_reduce_sum_keepdims_example",
     "test_reduce_sum_keepdims_random",
     "test_reduce_sum_negative_axes_keepdims_example",
-    "test_reduce_sum_square_default_axes_keepdims_example",
-    "test_reduce_sum_square_default_axes_keepdims_random",
     "test_reduce_sum_square_do_not_keepdims_example",
     "test_reduce_sum_square_do_not_keepdims_example_expanded",
     "test_reduce_sum_square_do_not_keepdims_random",
@@ -653,7 +647,6 @@
     "test_pow_types_float32_int64",
     "test_pow_types_float32_uint32",
     "test_pow_types_float32_uint64",
-    "test_qlinearconv",
     "test_qlinearmatmul_2D_int8_float16",
     "test_qlinearmatmul_2D_int8_float32",
     "test_qlinearmatmul_3D_int8_float16",
@@ -675,7 +668,9 @@
     "test_reduce_sum_empty_axes_input_noop_example",
     "test_reduce_sum_empty_set_non_reduced_axis_zero",
     "test_reduce_sum_negative_axes_keepdims_random",
+    "test_reduce_sum_square_default_axes_keepdims_example",
     "test_reduce_sum_square_default_axes_keepdims_example_expanded",
+    "test_reduce_sum_square_default_axes_keepdims_random",
     "test_reduce_sum_square_default_axes_keepdims_random_expanded",
     "test_shape_end_1",
     "test_shape_start_1",

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
@@ -59,8 +59,6 @@
     "test_bernoulli_double",
     "test_bernoulli_seed",
     "test_bernoulli_seed_expanded",
-    "test_blackmanwindow",
-    "test_blackmanwindow_symmetric",
     "test_cast_FLOAT16_to_FLOAT8E4M3FN",
     "test_cast_FLOAT16_to_FLOAT8E4M3FNUZ",
     "test_cast_FLOAT16_to_FLOAT8E5M2",
@@ -192,8 +190,6 @@
     "test_gru_with_initial_bias",
     "test_hammingwindow",
     "test_hammingwindow_symmetric",
-    "test_hannwindow",
-    "test_hannwindow_symmetric",
     "test_hardmax_axis_0",
     "test_hardmax_axis_1",
     "test_hardmax_axis_2",
@@ -378,8 +374,6 @@
     "test_reduce_sum_keepdims_example",
     "test_reduce_sum_keepdims_random",
     "test_reduce_sum_negative_axes_keepdims_example",
-    "test_reduce_sum_square_default_axes_keepdims_example",
-    "test_reduce_sum_square_default_axes_keepdims_random",
     "test_reduce_sum_square_do_not_keepdims_example",
     "test_reduce_sum_square_do_not_keepdims_example_expanded",
     "test_reduce_sum_square_do_not_keepdims_random",
@@ -660,7 +654,6 @@
     "test_pow_types_float32_int64",
     "test_pow_types_float32_uint32",
     "test_pow_types_float32_uint64",
-    "test_qlinearconv",
     "test_qlinearmatmul_2D_int8_float16",
     "test_qlinearmatmul_2D_int8_float32",
     "test_qlinearmatmul_3D_int8_float16",
@@ -687,7 +680,9 @@
     "test_reduce_sum_empty_axes_input_noop_example",
     "test_reduce_sum_empty_set_non_reduced_axis_zero",
     "test_reduce_sum_negative_axes_keepdims_random",
+    "test_reduce_sum_square_default_axes_keepdims_example",
     "test_reduce_sum_square_default_axes_keepdims_example_expanded",
+    "test_reduce_sum_square_default_axes_keepdims_random",
     "test_reduce_sum_square_default_axes_keepdims_random_expanded",
     "test_scatter_elements_with_negative_indices",
     "test_shape",

diff --git a/build_tools/pkgci/external_test_suite/onnx_gpu_vulkan.json b/build_tools/pkgci/external_test_suite/onnx_gpu_vulkan.json
@@ -74,8 +74,6 @@
     "test_bernoulli_double",
     "test_bernoulli_seed",
     "test_bernoulli_seed_expanded",
-    "test_blackmanwindow",
-    "test_blackmanwindow_symmetric",
     "test_cast_FLOAT16_to_FLOAT8E4M3FN",
     "test_cast_FLOAT16_to_FLOAT8E4M3FNUZ",
     "test_cast_FLOAT16_to_FLOAT8E5M2",
@@ -213,8 +211,6 @@
     "test_gru_with_initial_bias",
     "test_hammingwindow",
     "test_hammingwindow_symmetric",
-    "test_hannwindow",
-    "test_hannwindow_symmetric",
     "test_hardmax_axis_0",
     "test_hardmax_axis_1",
     "test_hardmax_axis_2",
@@ -409,8 +405,6 @@
     "test_reduce_sum_keepdims_example",
     "test_reduce_sum_keepdims_random",
     "test_reduce_sum_negative_axes_keepdims_example",
-    "test_reduce_sum_square_default_axes_keepdims_example",
-    "test_reduce_sum_square_default_axes_keepdims_random",
     "test_reduce_sum_square_do_not_keepdims_example",
     "test_reduce_sum_square_do_not_keepdims_example_expanded",
     "test_reduce_sum_square_do_not_keepdims_random",
@@ -740,7 +734,9 @@
     "test_reduce_sum_empty_axes_input_noop_example",
     "test_reduce_sum_empty_set_non_reduced_axis_zero",
     "test_reduce_sum_negative_axes_keepdims_random",
+    "test_reduce_sum_square_default_axes_keepdims_example",
     "test_reduce_sum_square_default_axes_keepdims_example_expanded",
+    "test_reduce_sum_square_default_axes_keepdims_random",
     "test_reduce_sum_square_default_axes_keepdims_random_expanded",
     "test_scatter_elements_with_negative_indices",
     "test_shape_end_1",

diff --git a/third_party/torch-mlir b/third_party/torch-mlir