update to torch nightly

1990a343 · rusty1s · 7318c1b8 · 1990a343
Commit 1990a343 authored Jul 18, 2019 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

test/utils.py test/utils.py +2 -0

No files found.
--- a/test/utils.py
+++ b/test/utils.py
@@ -4,6 +4,8 @@ from torch.testing import get_all_dtypes
 dtypes = get_all_dtypes()
 dtypes.remove(torch.half)
 dtypes.remove(torch.bool)
+if torch.bfloat16 in dtypes:
+    dtypes.remove(torch.bfloat16)
 grad_dtypes = [torch.float, torch.double]