"...git@developer.sourcefind.cn:kecinstone/2024-pra-vllm.git" did not exist on "58df2883cb3d3813e1d09ba691744773d9dcae58"
Commit ef65dd33 authored by zhaochao's avatar zhaochao
Browse files

[DCU] Skip alpha non-1 tests


Signed-off-by: default avatarzhaochao <zhaochao1@sugon.com>
parent 3d36696b
......@@ -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]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment