Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
bitsandbytes
Commits
24d9139e
Commit
24d9139e
authored
Jul 14, 2025
by
Egor Krivov
Browse files
Removed cpu
parent
36f5c4f4
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
tests/helpers.py
tests/helpers.py
+1
-1
No files found.
tests/helpers.py
View file @
24d9139e
...
@@ -21,7 +21,7 @@ BOOLEAN_TUPLES = list(product(TRUE_FALSE, repeat=2)) # all combinations of (boo
...
@@ -21,7 +21,7 @@ BOOLEAN_TUPLES = list(product(TRUE_FALSE, repeat=2)) # all combinations of (boo
def
get_available_devices
(
no_cpu
=
False
):
def
get_available_devices
(
no_cpu
=
False
):
if
"BNB_TEST_DEVICE"
in
os
.
environ
:
if
"BNB_TEST_DEVICE"
in
os
.
environ
:
# If the environment variable is set, use it directly.
# 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
[]
devices
=
[]
if
HIP_ENVIRONMENT
else
[
"cpu"
]
if
not
no_cpu
else
[]
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment