Skip to content

Commit 24d9139

Browse files
committed
Removed cpu
1 parent 36f5c4f commit 24d9139

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/helpers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
def get_available_devices(no_cpu=False):
2222
if "BNB_TEST_DEVICE" in os.environ:
2323
# If the environment variable is set, use it directly.
24-
return [os.environ["BNB_TEST_DEVICE"]]
24+
return [d for d in os.environ["BNB_TEST_DEVICE"] if d.lower() != "cpu"]
2525

2626
devices = [] if HIP_ENVIRONMENT else ["cpu"] if not no_cpu else []
2727

0 commit comments

Comments
 (0)