from __future__ import annotations

from .base import DeviceBackend


class MusaBackend(DeviceBackend):
    name = "musa"
    visibility_env_keys = ("MUSA_VISIBLE_DEVICES",)
    torch_namespace = "musa"
    flaggems_vendor = "musa"
    supports_cuda_graph = False
    supports_mm_cluster_fix = False
    supports_triton_hooks = True
    # Device count comes from torch.musa or MUSA_VISIBLE_DEVICES. Do not guess
    # a vendor CLI here; confirm the real listing tool before adding it.
    list_command = None