utils.py

import torch


@torch.jit.unused
def is_torch_compiling():
    try:
        return torch.compiler.is_compiling()
    except Exception:
        try:
            import torch._dynamo as dynamo  # noqa: F401

            return dynamo.is_compiling()
        except Exception:
            return False