You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
553 B
23 lines
553 B
5 months ago
|
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()
|