"""Torch 设备选择、类名归一化（子进程与 actionformer_gated 共用）。"""

from __future__ import annotations

import sys


def norm_product_name(name: str) -> str:
    s = (name or "").strip()
    if s == "一次性医用垫单":
        return "一次性使用手术单（一次性医用垫单）"
    return s


def resolve_inference_device(explicit: str) -> str | None:
    configured = (explicit or "").strip()
    if configured:
        return configured
    try:
        import torch
    except Exception:
        return None
    # 有 NVIDIA 时始终优先 CUDA（含 Linux / 少数非 Mac 环境）；Mac 无可用 CUDA 时再试 MPS。
    if torch.cuda.is_available():
        return "cuda:0"
    if sys.platform == "darwin" and torch.backends.mps.is_available():
        return "mps"
    return None