We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 36f5c4f commit 24d9139Copy full SHA for 24d9139
tests/helpers.py
@@ -21,7 +21,7 @@
21
def get_available_devices(no_cpu=False):
22
if "BNB_TEST_DEVICE" in os.environ:
23
# If the environment variable is set, use it directly.
24
- return [os.environ["BNB_TEST_DEVICE"]]
+ return [d for d in os.environ["BNB_TEST_DEVICE"] if d.lower() != "cpu"]
25
26
devices = [] if HIP_ENVIRONMENT else ["cpu"] if not no_cpu else []
27
0 commit comments