Skip cdouble dtype in pow test on TPU (#6772)

pytorch · Mar 19, 2024 · bd4420c · bd4420c
1 parent 7145a71
commit bd4420c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test/test_operations.py b/test/test_operations.py
@@ -1960,9 +1960,13 @@ def foo(x):
         torch.float32,
         torch.float64,
         torch.cfloat,
-        torch.cdouble,
     ]
 
+    if not _is_on_tpu():
+      test_dtypes += [
+          torch.cdouble,
+      ]
+
     for dtype in test_dtypes:
       test(dtype)