frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add __all__ to torch.utils submodules (#85331)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85331
Approved by: https://github.com/albanD
This commit is contained in:
anjali411
2022-09-27 12:30:22 +00:00
committed by PyTorch MergeBot
parent f64857189d
commit 0183c1e336
5 changed files with 10 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/utils/cpp_extension.py
View File

@ -70,7 +70,9 @@ CUDA_CLANG_VERSIONS = {
'11.7': ((6, 0, 0), (13, 0, 0)),
}
__all__ = ["get_default_build_root", "check_compiler_ok_for_platform", "get_compiler_abi_compatibility_and_version", "BuildExtension",
"CppExtension", "CUDAExtension", "include_paths", "library_paths", "load", "load_inline", "is_ninja_available",
"verify_ninja_availability"]
# Taken directly from python stdlib < 3.9
# See https://github.com/pytorch/pytorch/issues/48617
def _nt_quote_args(args: Optional[List[str]]) -> List[str]: