pytorch/torch/backends/openmp/__init__.py
2024-06-08 18:36:48 +00:00

8 lines
157 B
Python

# mypy: allow-untyped-defs
import torch
def is_available():
r"""Return whether PyTorch is built with OpenMP support."""
return torch._C.has_openmp