Skip to content

Commit 227ba73

Browse files
committed
Update on "Add ops (addr) | feat(torchlib)"
----- - Mark nan==nan in assert_close - Fix aten_isfinite by allowing the exporter to do type promotion according to #563 and introduce `TFloatHighPrecision` for it - Fix typo in `TRealOrUInt8` [ghstack-poisoned]
2 parents a1d0fcb + d6a611e commit 227ba73

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

onnxscript/tests/function_libs/torch_lib/ops_test_data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -431,7 +431,7 @@ def _where_input_wrangler(
431431
TorchLibOpInfo("abs", core_ops.aten_abs_complex, complex=True),
432432
TorchLibOpInfo("acos", core_ops.aten_acos),
433433
TorchLibOpInfo("acosh", core_ops.aten_acosh),
434-
TorchLibOpInfo("add", core_ops.aten_add),
434+
TorchLibOpInfo("add", core_ops.aten_add, tolerance={torch.float16: (1e-3, 1e-3)}),
435435
TorchLibOpInfo("addbmm", core_ops.aten_addbmm),
436436
TorchLibOpInfo("addcdiv", core_ops.aten_addcdiv),
437437
TorchLibOpInfo("addcmul", core_ops.aten_addcmul),

0 commit comments

Comments
 (0)