scripts/decode_precision_policy.py

from __future__ import annotations


def torch_dtype_name(precision: str) -> str | None:
    if precision in {"orbitquant", "fp32"}:
        return None
    if precision == "bf16":
        return "bfloat16"
    raise ValueError(f"unknown precision: {precision}")