speculators.utils.util

def get_current_device() -> str:
    """Get the current accelerator device string (e.g. 'cuda:0', 'npu:0')."""
    acc = torch.accelerator.current_accelerator()
    if acc is None:
        return "cuda:0"
    device_idx = torch.accelerator.current_device_index()
    return f"{acc.type}:{device_idx}"

def is_npu_available() -> bool:
    """Detect Ascend NPU availability"""
    try:
        return is_torch_npu_available()
    except (ImportError, RuntimeError, ModuleNotFoundError, AttributeError):
        return False