Skip to content

Commit d5f77f1

Browse files
authored
mkdir triton package and move triton files (#4420)
### What this PR does / why we need it? mkdir triton package and move triton files - vLLM version: v0.11.0 - vLLM main: vllm-project/vllm@2918c1b Signed-off-by: shiyuan680 <917935075@qq.com>
1 parent 1b137d6 commit d5f77f1

File tree

7 files changed

+5
-4
lines changed

7 files changed

+5
-4
lines changed

vllm_ascend/ops/triton/__init__.py

Whitespace-only changes.

vllm_ascend/ops/triton/fla/__init__.py

Whitespace-only changes.
File renamed without changes.
File renamed without changes.

vllm_ascend/ops/triton/mamba/__init__.py

Whitespace-only changes.
File renamed without changes.

vllm_ascend/patch/worker/patch_triton.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,11 +3,12 @@
33
import vllm.model_executor.layers.fla.ops.layernorm_guard
44
import vllm.model_executor.layers.mamba.ops.causal_conv1d
55

6-
from vllm_ascend.ops.casual_conv1d import (causal_conv1d_fn,
7-
causal_conv1d_update_npu)
8-
from vllm_ascend.ops.fla import LayerNormFn, torch_chunk_gated_delta_rule
9-
from vllm_ascend.ops.sigmoid_gating import \
6+
from vllm_ascend.ops.triton.fla.fla import (LayerNormFn,
7+
torch_chunk_gated_delta_rule)
8+
from vllm_ascend.ops.triton.fla.sigmoid_gating import \
109
fused_recurrent_gated_delta_rule_fwd_kernel
10+
from vllm_ascend.ops.triton.mamba.casual_conv1d import (
11+
causal_conv1d_fn, causal_conv1d_update_npu)
1112

1213
vllm.model_executor.layers.mamba.ops.causal_conv1d.causal_conv1d_update = causal_conv1d_update_npu
1314
vllm.model_executor.layers.mamba.ops.causal_conv1d.causal_conv1d_fn = causal_conv1d_fn

0 commit comments

Comments
 (0)