# Copyright (c) 2026 DeepBeepMeep - part of WanGP, WanGP Community License 2.0 (see LICENSE.txt)
"""MPS compatibility helpers and local test scripts."""


def is_mps_available():
    import torch
    return hasattr(torch.backends, "mps") and torch.backends.mps.is_available()


def mps_device():
    import torch
    return torch.device("mps") if is_mps_available() else None


def mps_device_or(default):
    return mps_device() or default