23 lines
574 B
Python
23 lines
574 B
Python
|
import torch
|
||
|
|
|
||
|
|
|
||
|
|
def show() -> str:
|
||
|
|
"""
|
||
|
|
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() -> str:
|
||
|
|
"""Returns the CXX_FLAGS used when building PyTorch."""
|
||
|
|
return torch._C._cxx_flags()
|
||
|
|
|
||
|
|
|
||
|
|
def parallel_info() -> str:
|
||
|
|
r"""Returns detailed string with parallelization settings"""
|
||
|
|
return torch._C._parallel_info()
|