From 7322ab8dfd1a682b7765feb3f61429417a60eaee Mon Sep 17 00:00:00 2001 From: chenyu Date: Tue, 5 Dec 2023 00:04:08 -0500 Subject: [PATCH] onnx tests with different dtypes (#2612) --- test/external/external_test_onnx_backend.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/test/external/external_test_onnx_backend.py b/test/external/external_test_onnx_backend.py index c3059203b1..65b9ffa6cd 100644 --- a/test/external/external_test_onnx_backend.py +++ b/test/external/external_test_onnx_backend.py @@ -47,14 +47,16 @@ backend_test.exclude('test_reduce_prod_*') backend_test.exclude('test_adam_multiple_cpu') backend_test.exclude('test_nesterov_momentum_cpu') -# we only support float32 -backend_test.exclude('uint8') -backend_test.exclude('uint16') -backend_test.exclude('uint32') -backend_test.exclude('uint64') -backend_test.exclude('int8') -backend_test.exclude('int16') -backend_test.exclude('float64') +# about different dtypes +backend_test.exclude('int8') # OverflowError: cannot convert float infinity to integer + +if Device.DEFAULT in ["TORCH"]: + backend_test.exclude('uint16') + backend_test.exclude('uint32') + backend_test.exclude('uint64') +if Device.DEFAULT in ["METAL"]: + backend_test.exclude('float64') + backend_test.exclude('string') backend_test.exclude('test_pow_types_int*')