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

Revert D21717199: [pytorch][PR] Updates assertEqual to require atol and rtol, removes positional atol

Browse Source 
Test Plan: revert-hammer

Differential Revision:
D21717199

Original commit changeset: 9feb856f94ee

fbshipit-source-id: bfde9c39a5ce99f0ca6183a7dde703c65b7c8259
This commit is contained in:
Rohan Varma
2020-05-26 18:21:26 -07:00
committed by Facebook GitHub Bot
parent ba14a701dc
commit 63e545e0fe
29 changed files with 831 additions and 843 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/test_cpp_extensions_jit.py
View File

@ -146,10 +146,10 @@ class TestCppExtensionJIT(common.TestCase):
actual_arches = sorted(re.findall(r'sm_\d\d', output))
expected_arches = ['sm_' + xx for xx in expected_values]
self.assertEqual(actual_arches, expected_arches,
msg="Flags: {}, Actual: {}, Expected: {}\n"
"Stderr: {}\nOutput: {}".format(
flags, actual_arches, expected_arches,
err, output))
message="Flags: {}, Actual: {}, Expected: {}\n"
"Stderr: {}\nOutput: {}".format(
flags, actual_arches, expected_arches,
err, output))
temp_dir = tempfile.mkdtemp()
old_envvar = os.environ.get('TORCH_CUDA_ARCH_LIST', None)