24 lines
580 B
Python
24 lines
580 B
Python
# mypy: allow-untyped-defs
|
|
import torch
|
|
|
|
|
|
def show():
|
|
"""
|
|
Return a human-readable string with descriptions of the
|
|
configuration of PyTorch.
|
|
"""
|
|
return torch._C._show_config()
|
|
|
|
|
|
# TODO: In principle, we could provide more structured version/config
|
|
# information here. For now only CXX_FLAGS is exposed, as Timer
|
|
# uses them.
|
|
def _cxx_flags():
|
|
"""Returns the CXX_FLAGS used when building PyTorch."""
|
|
return torch._C._cxx_flags()
|
|
|
|
|
|
def parallel_info():
|
|
r"""Returns detailed string with parallelization settings"""
|
|
return torch._C._parallel_info()
|