frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add OpInfo for torch.equal and fix support for non-standard bools

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/79389

Approved by: https://github.com/mruberry
This commit is contained in:
Peter Bell
2022-06-20 19:58:41 +01:00
committed by PyTorch MergeBot
parent aa911efdeb
commit 9bf52f4be8
7 changed files with 48 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_ops_jit.py
View File

@ -107,7 +107,7 @@ class TestJit(JitCommonTestCase):
# Check traced forward, grad, and grad grad
# TODO: fix tracing here
supports_tracing = not has_fake_function
supports_tracing = op.supports_tracing and not has_fake_function
if op.assert_jit_shape_analysis:
self.assertTrue(supports_tracing)