update dtype for ROCm

kshitij12345 · kshitij12345 · commit ecbc5320bb09 · 2021-03-09T05:06:31.000-06:00
diff --git a/torch/testing/_internal/common_methods_invocations.py b/torch/testing/_internal/common_methods_invocations.py
@@ -1625,8 +1625,9 @@ def _make_tensor_helper(shape, low=None, high=None):
     UnaryUfuncInfo('bitwise_not',
                    ref=np.bitwise_not,
                    dtypes=integral_types_and(torch.bool),
-                   dtypesIfCPU=integral_types_and(torch.bool),
-                   dtypesIfCUDA=integral_types_and(torch.bool),
+                   dtypesIfCPU=None,
+                   dtypesIfCUDA=None,
+                   dtypesIfROCM=None,
                    supports_autograd=False),
     UnaryUfuncInfo('ceil',
                    ref=np.ceil,
