from __future__ import annotations

from .base import DeviceBackend


class AscendBackend(DeviceBackend):
    name = "ascend"
    visibility_env_keys = (
        "ASCEND_RT_VISIBLE_DEVICES",
        "ASCEND_VISIBLE_DEVICES",
    )
    torch_namespace = "npu"
    flaggems_vendor = "ascend"
    supports_cuda_graph = False
    supports_mm_cluster_fix = False
    supports_triton_hooks = True
    list_command = None