[onnx] Toggle `f64` support for `onnx` test suite (#18111)
Some of the `onnx` tests require `f64` support. Turn on support for
validating the end to end support.
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
index 890bec8..31ccc21 100644
--- a/build_tools/pkgci/external_test_suite/onnx_cpu_llvm_sync.json
+++ b/build_tools/pkgci/external_test_suite/onnx_cpu_llvm_sync.json
@@ -2,7 +2,7 @@
"config_name": "cpu_llvm_sync",
"iree_compile_flags": [
"--iree-hal-target-backends=llvm-cpu",
- "--iree-input-demote-f64-to-f32"
+ "--iree-input-demote-f64-to-f32=false"
],
"iree_run_module_flags": [
"--device=local-sync"
@@ -40,6 +40,7 @@
"onnx/node/generated/test_batchnorm_example_training_mode",
"onnx/node/generated/test_bernoulli_seed",
"onnx/node/generated/test_bernoulli_seed_expanded",
+ "onnx/node/generated/test_cast_DOUBLE_to_FLOAT16",
"onnx/node/generated/test_cast_FLOAT16_to_FLOAT8E4M3FN",
"onnx/node/generated/test_cast_FLOAT16_to_FLOAT8E4M3FNUZ",
"onnx/node/generated/test_cast_FLOAT16_to_FLOAT8E5M2",
@@ -76,6 +77,8 @@
"onnx/node/generated/test_cast_no_saturate_FLOAT_to_FLOAT8E4M3FNUZ",
"onnx/node/generated/test_cast_no_saturate_FLOAT_to_FLOAT8E5M2",
"onnx/node/generated/test_cast_no_saturate_FLOAT_to_FLOAT8E5M2FNUZ",
+ "onnx/node/generated/test_castlike_DOUBLE_to_FLOAT16",
+ "onnx/node/generated/test_castlike_DOUBLE_to_FLOAT16_expanded",
"onnx/node/generated/test_castlike_FLOAT8E4M3FNUZ_to_FLOAT",
"onnx/node/generated/test_castlike_FLOAT8E4M3FNUZ_to_FLOAT_expanded",
"onnx/node/generated/test_castlike_FLOAT8E4M3FN_to_FLOAT",
@@ -290,7 +293,9 @@
"onnx/node/generated/test_reduce_log_sum_desc_axes_expanded",
"onnx/node/generated/test_reduce_log_sum_empty_set",
"onnx/node/generated/test_reduce_log_sum_empty_set_expanded",
+ "onnx/node/generated/test_reduce_log_sum_exp_default_axes_keepdims_example",
"onnx/node/generated/test_reduce_log_sum_exp_default_axes_keepdims_example_expanded",
+ "onnx/node/generated/test_reduce_log_sum_exp_default_axes_keepdims_random",
"onnx/node/generated/test_reduce_log_sum_exp_default_axes_keepdims_random_expanded",
"onnx/node/generated/test_reduce_log_sum_exp_do_not_keepdims_example",
"onnx/node/generated/test_reduce_log_sum_exp_do_not_keepdims_example_expanded",
@@ -515,33 +520,17 @@
"onnx/node/generated/test_bernoulli_double",
"onnx/node/generated/test_bernoulli_double_expanded",
"onnx/node/generated/test_bernoulli_expanded",
- "onnx/node/generated/test_cast_DOUBLE_to_FLOAT",
- "onnx/node/generated/test_cast_DOUBLE_to_FLOAT16",
- "onnx/node/generated/test_cast_FLOAT16_to_DOUBLE",
"onnx/node/generated/test_cast_FLOAT_to_BFLOAT16",
- "onnx/node/generated/test_cast_FLOAT_to_DOUBLE",
- "onnx/node/generated/test_castlike_DOUBLE_to_FLOAT",
- "onnx/node/generated/test_castlike_DOUBLE_to_FLOAT16",
- "onnx/node/generated/test_castlike_DOUBLE_to_FLOAT16_expanded",
- "onnx/node/generated/test_castlike_DOUBLE_to_FLOAT_expanded",
- "onnx/node/generated/test_castlike_FLOAT16_to_DOUBLE",
- "onnx/node/generated/test_castlike_FLOAT16_to_DOUBLE_expanded",
"onnx/node/generated/test_castlike_FLOAT_to_BFLOAT16",
"onnx/node/generated/test_castlike_FLOAT_to_BFLOAT16_expanded",
- "onnx/node/generated/test_castlike_FLOAT_to_DOUBLE",
- "onnx/node/generated/test_castlike_FLOAT_to_DOUBLE_expanded",
"onnx/node/generated/test_constantofshape_float_ones",
"onnx/node/generated/test_constantofshape_int_shape_zero",
"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_eyelike_with_dtype",
"onnx/node/generated/test_gather_elements_negative_indices",
"onnx/node/generated/test_gridsample_nearest",
"onnx/node/generated/test_gridsample_nearest_align_corners_0_additional_1",
"onnx/node/generated/test_gridsample_nearest_align_corners_1_additional_1",
- "onnx/node/generated/test_max_float64",
- "onnx/node/generated/test_min_float64",
"onnx/node/generated/test_mod_mixed_sign_int16",
"onnx/node/generated/test_mod_mixed_sign_int32",
"onnx/node/generated/test_mod_mixed_sign_int64",
@@ -559,8 +548,6 @@
"onnx/node/generated/test_qlinearmatmul_3D_int8_float32",
"onnx/node/generated/test_qlinearmatmul_3D_uint8_float16",
"onnx/node/generated/test_qlinearmatmul_3D_uint8_float32",
- "onnx/node/generated/test_reduce_log_sum_exp_default_axes_keepdims_example",
- "onnx/node/generated/test_reduce_log_sum_exp_default_axes_keepdims_random",
"onnx/node/generated/test_reduce_max_empty_set",
"onnx/node/generated/test_reduce_min_empty_set",
"onnx/node/generated/test_reduce_sum_empty_axes_input_noop",
diff --git a/build_tools/pkgci/external_test_suite/onnx_gpu_cuda.json b/build_tools/pkgci/external_test_suite/onnx_gpu_cuda.json
index fc3f75f..721436d 100644
--- a/build_tools/pkgci/external_test_suite/onnx_gpu_cuda.json
+++ b/build_tools/pkgci/external_test_suite/onnx_gpu_cuda.json
@@ -2,7 +2,7 @@
"config_name": "gpu_cuda_t4",
"iree_compile_flags": [
"--iree-hal-target-backends=cuda",
- "--iree-input-demote-f64-to-f32"
+ "--iree-input-demote-f64-to-f32=false"
],
"iree_run_module_flags": [
"--device=cuda"
@@ -518,33 +518,16 @@
"onnx/node/generated/test_bernoulli_double",
"onnx/node/generated/test_bernoulli_double_expanded",
"onnx/node/generated/test_bernoulli_expanded",
- "onnx/node/generated/test_cast_DOUBLE_to_FLOAT",
- "onnx/node/generated/test_cast_DOUBLE_to_FLOAT16",
- "onnx/node/generated/test_cast_FLOAT16_to_DOUBLE",
"onnx/node/generated/test_cast_FLOAT_to_BFLOAT16",
- "onnx/node/generated/test_cast_FLOAT_to_DOUBLE",
- "onnx/node/generated/test_castlike_DOUBLE_to_FLOAT",
- "onnx/node/generated/test_castlike_DOUBLE_to_FLOAT16",
- "onnx/node/generated/test_castlike_DOUBLE_to_FLOAT16_expanded",
- "onnx/node/generated/test_castlike_DOUBLE_to_FLOAT_expanded",
- "onnx/node/generated/test_castlike_FLOAT16_to_DOUBLE",
- "onnx/node/generated/test_castlike_FLOAT16_to_DOUBLE_expanded",
"onnx/node/generated/test_castlike_FLOAT_to_BFLOAT16",
"onnx/node/generated/test_castlike_FLOAT_to_BFLOAT16_expanded",
- "onnx/node/generated/test_castlike_FLOAT_to_DOUBLE",
- "onnx/node/generated/test_castlike_FLOAT_to_DOUBLE_expanded",
"onnx/node/generated/test_constantofshape_float_ones",
"onnx/node/generated/test_constantofshape_int_shape_zero",
"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_eyelike_with_dtype",
"onnx/node/generated/test_gridsample_nearest",
"onnx/node/generated/test_gridsample_nearest_align_corners_0_additional_1",
"onnx/node/generated/test_gridsample_nearest_align_corners_1_additional_1",
- "onnx/node/generated/test_max_float64",
- "onnx/node/generated/test_min_float64",
- "onnx/node/generated/test_mod_mixed_sign_float64",
"onnx/node/generated/test_mod_mixed_sign_int16",
"onnx/node/generated/test_mod_mixed_sign_int32",
"onnx/node/generated/test_mod_mixed_sign_int64",
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
index 238d7af..421b957 100644
--- a/build_tools/pkgci/external_test_suite/onnx_gpu_rocm_rdna3.json
+++ b/build_tools/pkgci/external_test_suite/onnx_gpu_rocm_rdna3.json
@@ -3,7 +3,7 @@
"iree_compile_flags": [
"--iree-hal-target-backends=rocm",
"--iree-rocm-target-chip=gfx1100",
- "--iree-input-demote-f64-to-f32"
+ "--iree-input-demote-f64-to-f32=false"
],
"iree_run_module_flags": [
"--device=hip"
@@ -513,11 +513,7 @@
"onnx/node/generated/test_bernoulli_double",
"onnx/node/generated/test_bernoulli_double_expanded",
"onnx/node/generated/test_bernoulli_expanded",
- "onnx/node/generated/test_cast_DOUBLE_to_FLOAT",
- "onnx/node/generated/test_cast_DOUBLE_to_FLOAT16",
- "onnx/node/generated/test_cast_FLOAT16_to_DOUBLE",
"onnx/node/generated/test_cast_FLOAT_to_BFLOAT16",
- "onnx/node/generated/test_cast_FLOAT_to_DOUBLE",
"onnx/node/generated/test_castlike_BFLOAT16_to_FLOAT",
"onnx/node/generated/test_castlike_BFLOAT16_to_FLOAT_expanded",
"onnx/node/generated/test_castlike_DOUBLE_to_FLOAT",
@@ -538,7 +534,6 @@
"onnx/node/generated/test_constantofshape_int_shape_zero",
"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_expand_dim_changed",
"onnx/node/generated/test_expand_dim_unchanged",
"onnx/node/generated/test_eyelike_populate_off_main_diagonal",
@@ -549,9 +544,6 @@
"onnx/node/generated/test_gridsample_nearest_align_corners_0_additional_1",
"onnx/node/generated/test_gridsample_nearest_align_corners_1_additional_1",
"onnx/node/generated/test_gridsample_zeros_padding",
- "onnx/node/generated/test_max_float64",
- "onnx/node/generated/test_min_float64",
- "onnx/node/generated/test_mod_mixed_sign_float64",
"onnx/node/generated/test_mod_mixed_sign_int16",
"onnx/node/generated/test_mod_mixed_sign_int32",
"onnx/node/generated/test_mod_mixed_sign_int64",
