[CI] Add Triton Ascend in CI (#4921)
Add triton-ascend in UT and e2e
- vLLM version: v0.12.0
- vLLM main:
ad32e3e19c
---------
Signed-off-by: Meihan-chen <jcccx.cmh@gmail.com>
This commit is contained in:
@@ -1,3 +1,4 @@
|
||||
import sys
|
||||
from unittest.mock import MagicMock
|
||||
|
||||
import torch
|
||||
@@ -5,6 +6,10 @@ from vllm.v1.attention.backends.utils import AttentionCGSupport
|
||||
|
||||
from tests.ut.base import TestBase
|
||||
from vllm_ascend.attention.attention_v1 import AscendAttentionState
|
||||
|
||||
if 'torch_npu._inductor' not in sys.modules:
|
||||
sys.modules['torch_npu._inductor'] = MagicMock()
|
||||
|
||||
from vllm_ascend.attention.sfa_v1 import (AscendSFABackend, AscendSFAImpl,
|
||||
AscendSFAMetadata,
|
||||
AscendSFAMetadataBuilder)
|
||||
|
||||
Reference in New Issue
Block a user