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

Add __all__ to torch.distributed, futures, fx, nn, package, benchmark submodules (#80520)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/80520
Approved by: https://github.com/rohan-varma
This commit is contained in:
anjali411
2022-07-07 18:21:06 +00:00
committed by PyTorch MergeBot
parent 81ca2ff353
commit 4bf076e964
21 changed files with 48 additions and 141 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
torch/utils/hipify/hipify_python.py
View File

@ -46,6 +46,12 @@ HIPIFY_FINAL_RESULT: HipifyFinalResult = {}
to their actual types."""
PYTORCH_TEMPLATE_MAP = {"Dtype": "scalar_t", "T": "scalar_t"}
__all__ = ['InputError', 'openf', 'bcolors', 'GeneratedFileCleaner', 'match_extensions', 'matched_files_iter',
'preprocess_file_and_save_result', 'compute_stats', 'add_dim3', 'processKernelLaunches', 'find_closure_group',
'find_bracket_group', 'find_parentheses_group', 'replace_math_functions', 'hip_header_magic', 'replace_extern_shared',
'get_hip_file_path', 'is_out_of_place', 'is_pytorch_file', 'is_cusparse_file', 'is_caffe2_gpu_file',
'is_caffe2_gpu_file', 'Trie', 'preprocessor', 'file_specific_replacement', 'file_add_header',
'fix_static_global_kernels', 'extract_arguments', 'str2bool', 'hipify']
class InputError(Exception):
# Exception raised for errors in the input.