update to torch nightly

cb6c8b71 · rusty1s · 0a9f541c · cb6c8b71
Commit cb6c8b71 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
@@ -6,6 +6,8 @@ dtypes = get_all_dtypes()
 dtypes.remove(torch.half)
 dtypes.remove(torch.short)  # PyTorch scatter does not work on short types.
 dtypes.remove(torch.bool)
+if torch.bfloat16 in dtypes:
+    dtypes.remove(torch.bfloat16)
 grad_dtypes = [torch.float, torch.double]