Skip to content

Commit 98a4d21

Browse files
committed
fix triton
Signed-off-by: wxsIcey <[email protected]>
1 parent 20568f7 commit 98a4d21

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

vllm_ascend/ops/__init__.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,11 +16,15 @@
1616
#
1717

1818
import torch
19+
from vllm.triton_utils import HAS_TRITON
1920

2021
import vllm_ascend.ops.fused_moe.fused_moe # noqa
2122
import vllm_ascend.ops.layernorm # noqa
2223
import vllm_ascend.ops.register_custom_ops # noqa
23-
import vllm_ascend.ops.triton.linearnorm.split_qkv_rmsnorm_rope # noqa
24+
25+
if HAS_TRITON:
26+
import vllm_ascend.ops.triton.linearnorm.split_qkv_rmsnorm_rope # noqa
27+
2428
import vllm_ascend.ops.vocab_parallel_embedding # noqa
2529
from vllm_ascend.ops.activation import AscendQuickGELU, AscendSiluAndMul
2630
from vllm_ascend.ops.rotary_embedding import (

0 commit comments

Comments
 (0)