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
70bacda7
Commit
70bacda7
authored
May 21, 2025
by
Matthew Douglas
Browse files
Update test for L40S
parent
cdcae8d3
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
7 additions
and
1 deletion
+7
-1
tests/test_functional.py
tests/test_functional.py
+7
-1
No files found.
tests/test_functional.py
View file @
70bacda7
...
@@ -525,6 +525,12 @@ class TestIGEMMFunctional:
...
@@ -525,6 +525,12 @@ class TestIGEMMFunctional:
# print(mean(errs2))
# print(mean(errs2))
# print(mean(relerrs2))
# print(mean(relerrs2))
assert
mean
(
errs
)
<
0.015
assert
mean
(
errs
)
<
0.015
# There's a higher relerr on L40S with torch 2.4+cu118.
is_sm89
=
torch
.
cuda
.
get_device_capability
()
==
(
8
,
9
)
if
torch
.
version
.
cuda
==
"11.8"
and
is_sm89
and
torch
.
__version__
<
(
2
,
5
):
assert
mean
(
relerrs
)
<
0.41
else
:
assert
mean
(
relerrs
)
<
0.3
assert
mean
(
relerrs
)
<
0.3
@
pytest
.
mark
.
parametrize
(
"dim1"
,
[
1
,
64
],
ids
=
id_formatter
(
"dim1"
))
@
pytest
.
mark
.
parametrize
(
"dim1"
,
[
1
,
64
],
ids
=
id_formatter
(
"dim1"
))
...
...
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