Skip to content

Commit 1cb7b10

Browse files
authored
[0.9.1][Fix] Removes explicit ATB extension registration (#1921)
### What this PR does / why we need it? Since the ATB extension registration has now been fixed in torch_npu, we have removed the invocation of this private method. ### Does this PR introduce _any_ user-facing change? None. ### How was this patch tested? No need for further testing. Signed-off-by: Yizhou Liu <liu_yizhou@outlook.com>
1 parent cff4214 commit 1cb7b10

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

vllm_ascend/worker/worker_v1.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@
2222
import torch
2323
import torch.nn as nn
2424
import torch_npu
25-
from torch_npu.op_plugin.atb._atb_ops import _register_atb_extensions
2625
from vllm import envs
2726
from vllm.config import VllmConfig
2827
from vllm.distributed import (ensure_model_parallel_initialized,
@@ -70,7 +69,6 @@ def __init__(
7069
# Register ops when worker init.
7170
from vllm_ascend import ops
7271
ops.register_dummy_fusion_op()
73-
_register_atb_extensions()
7472
# init ascend config
7573
init_ascend_config(vllm_config)
7674

0 commit comments

Comments
 (0)