Skip to content

Commit d6a611e

Browse files
committed
Update base for 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]
1 parent 5b2200a commit d6a611e

File tree

1 file changed

+1
-8
lines changed

1 file changed

+1
-8
lines changed

onnxscript/tests/function_libs/torch_lib/ops_test_data.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -431,14 +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(
435-
"add",
436-
core_ops.aten_add,
437-
).xfail(
438-
dtypes=[torch.float16],
439-
reason="fixme: float16 failed, tensor-likes are not close for FullGraph mode. https://github.com/microsoft/onnxruntime/issues/15977",
440-
test_class_name="TestOutputConsistencyFullGraph",
441-
),
434+
TorchLibOpInfo("add", core_ops.aten_add, tolerance={torch.float16: (1e-3, 1e-3)}),
442435
TorchLibOpInfo("addbmm", core_ops.aten_addbmm),
443436
TorchLibOpInfo("addcdiv", core_ops.aten_addcdiv),
444437
TorchLibOpInfo("addcmul", core_ops.aten_addcmul),

0 commit comments

Comments
 (0)