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
68d851d1
Commit
68d851d1
authored
Oct 20, 2025
by
zhaochao
Browse files
[DCU] Skip alpha non-1 tests
Signed-off-by:
zhaochao
<
zhaochao1@sugon.com
>
parent
98d282c9
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
1 deletion
+5
-1
tests/pytorch/test_fusible_ops.py
tests/pytorch/test_fusible_ops.py
+5
-1
No files found.
tests/pytorch/test_fusible_ops.py
View file @
68d851d1
...
...
@@ -37,6 +37,7 @@ from transformer_engine.pytorch.tensor.float8_tensor import (
from
transformer_engine.pytorch.tensor.mxfp8_tensor
import
MXFP8Tensor
,
MXFP8Quantizer
from
transformer_engine.pytorch.utils
import
is_bf16_compatible
import
transformer_engine_torch
as
tex
from
torch.utils.cpp_extension
import
IS_HIP_EXTENSION
# Import utility functions
_current_file
=
pathlib
.
Path
(
__file__
).
resolve
()
...
...
@@ -2040,6 +2041,8 @@ class TestFusedOps:
)
->
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
]
...
...
@@ -2336,7 +2339,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