Skip to content

Commit 3017d88

Browse files
authored
[tests] fix autoencoderdc ml training fix. (#14129)
fix autoencoderdc ml training fix.
1 parent a7166ce commit 3017d88

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

tests/models/testing_utils/training.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -210,7 +210,9 @@ def test_mixed_precision_training(self):
210210

211211
# Test with bfloat16
212212
if torch.device(torch_device).type != "cpu":
213-
if torch.device(torch_device).type == "cuda" and not torch.cuda.is_bf16_supported():
213+
if torch.device(torch_device).type == "cuda" and not torch.cuda.is_bf16_supported(
214+
including_emulation=False
215+
):
214216
pytest.skip("bfloat16 training is not supported on this GPU.")
215217
else:
216218
model.zero_grad()

0 commit comments

Comments
 (0)