Commit 24d9139e authored by Egor Krivov's avatar Egor Krivov
Browse files

Removed cpu

parent 36f5c4f4
......@@ -21,7 +21,7 @@ BOOLEAN_TUPLES = list(product(TRUE_FALSE, repeat=2)) # all combinations of (boo
def get_available_devices(no_cpu=False):
if "BNB_TEST_DEVICE" in os.environ:
# If the environment variable is set, use it directly.
return [os.environ["BNB_TEST_DEVICE"]]
return [d for d in os.environ["BNB_TEST_DEVICE"] if d.lower() != "cpu"]
devices = [] if HIP_ENVIRONMENT else ["cpu"] if not no_cpu else []
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment