Skip to content

Commit 398c46c

Browse files
wanfengcxzclaude
andcommitted
fix: update import path for init_device_properties_triton
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent 06031f9 commit 398c46c

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

lmdeploy/pytorch/backends/dlinfer/ascend/op_backend.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -451,8 +451,8 @@ def build_graph_runner(model: torch.nn.Module, model_config: ModelConfig, cache_
451451
def init():
452452
"""Initialize Ascend backend."""
453453
try:
454+
from dlinfer.vendor.ascend.triton_ops.triton_utils import init_device_properties_triton
454455
from torch_npu.contrib import transfer_to_npu # noqa: F401
455-
from dlinfer.vendor.ascend.triton_ops.fla.triton_utils import init_device_properties_triton
456456
init_device_properties_triton()
457457
except ImportError:
458458
logger.warning('Failed to import torch_npu. Please make sure torch_npu is installed correctly. '

0 commit comments

Comments
 (0)