We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a7166ce commit 3017d88Copy full SHA for 3017d88
1 file changed
tests/models/testing_utils/training.py
@@ -210,7 +210,9 @@ def test_mixed_precision_training(self):
210
211
# Test with bfloat16
212
if torch.device(torch_device).type != "cpu":
213
- if torch.device(torch_device).type == "cuda" and not torch.cuda.is_bf16_supported():
+ if torch.device(torch_device).type == "cuda" and not torch.cuda.is_bf16_supported(
214
+ including_emulation=False
215
+ ):
216
pytest.skip("bfloat16 training is not supported on this GPU.")
217
else:
218
model.zero_grad()
0 commit comments