Fixed issue with supported devices and dtypes on the gpu by adding gp…

…u back to the valid devices (ivy-llc#10721)
vedpatwardhan · Feb 21, 2023 · b2d19eb · b2d19eb
1 parent f45290f
commit b2d19eb
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/ivy/__init__.py b/ivy/__init__.py
@@ -311,9 +311,9 @@ class Node(str):
 # ToDo: add gpu and tpu for valid devices when we test for them
 all_devices = ("cpu", "gpu", "tpu")
 
-valid_devices = ("cpu",)
+valid_devices = ("cpu", "gpu")
 
-invalid_devices = ("gpu", "tpu")
+invalid_devices = ("tpu",)
 
 # data types as string (to be used by Dtype classes)
 # any changes here should also be reflected in the data type initialisation underneath

diff --git a/ivy/functional/backends/jax/__init__.py b/ivy/functional/backends/jax/__init__.py
@@ -58,9 +58,9 @@
 
 
 # devices
-valid_devices = ("cpu",)
+valid_devices = ("cpu", "gpu")
 
-invalid_devices = ("gpu", "tpu")
+invalid_devices = ("tpu",)
 
 
 # native data types

diff --git a/ivy/functional/backends/tensorflow/__init__.py b/ivy/functional/backends/tensorflow/__init__.py
@@ -29,9 +29,9 @@
 
 
 # devices
-valid_devices = ("cpu",)
+valid_devices = ("cpu", "gpu")
 
-invalid_devices = ("gpu", "tpu")
+invalid_devices = ("tpu",)
 
 
 # native data types

diff --git a/ivy/functional/backends/torch/__init__.py b/ivy/functional/backends/torch/__init__.py
@@ -25,9 +25,9 @@
 
 
 # devices
-valid_devices = ("cpu",)
+valid_devices = ("cpu", "gpu")
 
-invalid_devices = ("gpu", "tpu")
+invalid_devices = ("tpu",)
 
 
 # native data types