mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages 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
commit 7322ab8dfd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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*')