Searched refs:is_pytorch_extension (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/utils/hipify/ |
| H A D | hipify_python.py | 155 is_pytorch_extension: bool = False) -> Iterator[str]:
185 if not is_pytorch_extension: # for pytorch extensions, consider all files
200 is_pytorch_extension: bool,
207 hip_clang_launch, is_pytorch_extension, clean_ctx, show_progress)
539 def get_hip_file_path(rel_filepath, is_pytorch_extension=False): argument
546 if not is_pytorch_extension and not is_out_of_place(rel_filepath):
602 if not is_pytorch_extension and dirpath == orig_dirpath:
605 if is_pytorch_extension and dirpath == orig_dirpath and (root + ext) == orig_filename:
812 is_pytorch_extension: bool,
838 …path.abspath(os.path.join(output_directory, get_hip_file_path(rel_filepath, is_pytorch_extension)))
[all …]
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | cpp_extension.py | 1107 is_pytorch_extension=True,
1710 is_pytorch_extension=True,
|