mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
Pull Request resolved: https://github.com/pytorch/pytorch/pull/143320 Approved by: https://github.com/aorenste ghstack dependencies: #143319
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()
|