github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-10 23:48:01 -05:00
Code Issues Packages Projects Releases Wiki Activity

onnx tests with different dtypes (#2612)

Browse Source
This commit is contained in:
chenyu
2023-12-05 00:04:08 -05:00
committed by GitHub
parent f12bcccb87
commit 7322ab8dfd
1 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
test/external/external_test_onnx_backend.py vendored
View File

@@ -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*')