geohot/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Actions Activity

onnx fmod (#9307)

Browse Source
This commit is contained in:
chenyu
2025-02-28 14:09:22 -05:00
committed by GitHub
parent 7c7db78feb
commit 38d7aae3b7
2 changed files with 2 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extra/onnx.py
+1 -1
View File
@@ -317,7 +317,7 @@ def get_onnx_ops():
def BitwiseXor(x:Tensor,y:Tensor): return x ^ y
def BitwiseNot(x:Tensor): return ~x
def Mod(x:Tensor,y:Tensor,fmod=0):
if fmod != 0: raise NotImplementedError("float mod is not supported")
if fmod: return x - x.div(y, rounding_mode="trunc") * y
return x % y
# ***** Casting Ops *****
test/external/external_test_onnx_backend.py
Vendored
+1 -6
View File
@@ -77,6 +77,7 @@ backend_test.exclude('cast_no_saturate')
backend_test.exclude('test_dequantizelinear_e4m3fn_float16_cpu')
backend_test.exclude('test_max_float16_cpu')
backend_test.exclude('test_min_float16_cpu')
backend_test.exclude('test_mod_mixed_sign_float16_cpu')
backend_test.exclude('test_dequantizelinear_int4_cpu')
backend_test.exclude('test_dequantizelinear_uint4_cpu')
@@ -99,12 +100,6 @@ backend_test.exclude('test_nonzero_*')
backend_test.exclude('test_pow_types_int32_int32_cpu')
backend_test.exclude('test_pow_types_int64_int64_cpu')
# no support for fmod
backend_test.exclude('test_mod_int64_fmod_cpu')
backend_test.exclude('test_mod_mixed_sign_float16_cpu')
backend_test.exclude('test_mod_mixed_sign_float32_cpu')
backend_test.exclude('test_mod_mixed_sign_float64_cpu')
# no boolean ops (2d, 3d, 4d)
backend_test.exclude('test_bitshift_*')