OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

ci fix (#67826)

Browse Source 
Summary:
Fixes #{issue number}

Pull Request resolved: https://github.com/pytorch/pytorch/pull/67826

Reviewed By: Chillee

Differential Revision: D32164770

Pulled By: mruberry

fbshipit-source-id: c1de7e6db6d0cb1761388f1ea0178dbff3fe6dc8
This commit is contained in:
Mike Ruberry 2021-11-04 00:14:57 -07:00 committed by Facebook GitHub Bot
parent 1baed45c6b
commit b8d365ca3a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_jit_autocast.py
View File

@ -10,7 +10,7 @@ from torch.testing._internal.common_cuda import TEST_CUDA
from torch.testing._internal.common_utils import run_tests
from torch.testing import FileCheck
TEST_BFLOAT16 = torch.cuda.is_bf16_supported()
TEST_BFLOAT16 = TEST_CUDA and torch.cuda.is_bf16_supported()
class TestAutocast(JitTestCase):
def setUp(self):