EngineX/xc-llm-ascend
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[CI]repair ci for custom op (#8455)

Browse Source 
### What this PR does / why we need it?
repair ci for custom op nightly

Signed-off-by: ZT-AIA <1028681969@qq.com>
This commit is contained in:
ZT-AIA
2026-04-20 17:51:37 +08:00
committed by GitHub
parent dd7e08c6db
commit 3db5048d74
3 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/e2e/nightly/single_node/ops/singlecard_ops/test_mla_preprocess.py
View File

@@ -8,7 +8,9 @@ from vllm_ascend.utils import enable_custom_op
enable_custom_op()
@pytest.mark.skip(
reason="Failure of an individual operator use case causes failures of other operators."
)
@pytest.mark.parametrize("cache_mode", ["krope_ctkv", "nzcache"])
@torch.inference_mode()
def test_mla_preprocess_kernel(cache_mode: str):

5
tests/e2e/nightly/single_node/ops/singlecard_ops/test_mla_preprocess_nq.py
View File

@@ -1,5 +1,6 @@
import gc
import pytest
import torch
import torch_npu
@@ -7,7 +8,9 @@ from vllm_ascend.utils import enable_custom_op
enable_custom_op()
@pytest.mark.skip(
reason="Failure of an individual operator use case causes failures of other operators."
)
@torch.inference_mode()
def test_mla_preprocess_kernel():
token_num = 1

4
tests/e2e/nightly/single_node/ops/singlecard_ops/test_mla_preprocess_qdown.py
View File

@@ -8,7 +8,9 @@ from vllm_ascend.utils import enable_custom_op
enable_custom_op()
@pytest.mark.skip(
reason="Failure of an individual operator use case causes failures of other operators."
)
@pytest.mark.parametrize("cache_mode", ["krope_ctkv", "nzcache"])
@torch.inference_mode()
def test_mla_preprocess_kernel(cache_mode: str):