Skip testing torch.float16 in test_isnan (#55906)
Summary:
See https://github.com/pytorch/pytorch/issues/55905
Pull Request resolved: https://github.com/pytorch/pytorch/pull/55906
Reviewed By: walterddr
Differential Revision: D27737356
Pulled By: malfet
fbshipit-source-id: 39571cfe6f078af8bb7387ed459a5d0f2410bad1
diff --git a/test/test_jit_fuser_te.py b/test/test_jit_fuser_te.py
index 5b914e7..bcf8b97 100644
--- a/test/test_jit_fuser_te.py
+++ b/test/test_jit_fuser_te.py
@@ -1314,7 +1314,8 @@
dtypes = [
torch.bool,
torch.int,
- torch.float16,
+ # TODO: Add back when https://github.com/pytorch/pytorch/issues/55905 is closed
+ # torch.float16,
torch.float32,
torch.float64,
]
@@ -1383,7 +1384,8 @@
torch.int16,
torch.int32,
torch.int64,
- torch.float16,
+ # TODO: Add back when https://github.com/pytorch/pytorch/issues/55905 is closed
+ # torch.float16,
torch.float32,
torch.float64,
torch.bool,