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
TransformerEngine
Commits
ef65dd33
Commit
ef65dd33
authored
Nov 03, 2025
by
zhaochao
Browse files
[DCU] Skip alpha non-1 tests
Signed-off-by:
zhaochao
<
zhaochao1@sugon.com
>
parent
3d36696b
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
2 deletions
+4
-2
tests/pytorch/test_fusible_ops.py
tests/pytorch/test_fusible_ops.py
+4
-2
No files found.
tests/pytorch/test_fusible_ops.py
View file @
ef65dd33
...
...
@@ -2084,7 +2084,8 @@ class TestFusedOps:
quantized_weight
:
bool
=
False
,
)
->
None
:
"""Forward GEMM + scale + add"""
if
IS_HIP_EXTENSION
and
scale
!=
1
:
pytest
.
skip
(
"alpha must be 1.0 for hip"
)
# Make input and weight shapes consistent
out_features
,
in_features
=
weight_shape
in_shape
=
list
(
in_shape
)[:
-
1
]
+
[
in_features
]
...
...
@@ -2469,7 +2470,8 @@ class TestFusedOps:
quantized_weight
:
bool
=
False
,
)
->
None
:
"""Backward dgrad GEMM + scale"""
if
IS_HIP_EXTENSION
and
scale
!=
1
:
pytest
.
skip
(
"alpha must be 1.0 for hip"
)
# Make input and weight shapes consistent
out_features
,
in_features
=
weight_shape
in_shape
=
list
(
in_shape
)[:
-
1
]
+
[
in_features
]
...
...
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