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

[Inductor] Rename cpp_wrapper_cuda.py as cpp_wrapper_gpu.py (#135313)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/135313
Approved by: https://github.com/jansel, https://github.com/desertfire
ghstack dependencies: #135312
This commit is contained in:
xinan.lin
2024-09-10 03:13:14 -07:00
committed by PyTorch MergeBot
parent 13ee85ca5e
commit 16b37b309f
3 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tools/amd_build/build_amd.py
View File

@ -207,7 +207,7 @@ hipify_python.hipify(
ignores=ignores, ignores=ignores,
extra_files=[ extra_files=[
"torch/_inductor/codegen/cpp_wrapper_cpu.py", "torch/_inductor/codegen/cpp_wrapper_cpu.py",
"torch/_inductor/codegen/cpp_wrapper_cuda.py", "torch/_inductor/codegen/cpp_wrapper_gpu.py",
"torch/_inductor/codegen/wrapper.py", "torch/_inductor/codegen/wrapper.py",
], ],
out_of_place_only=args.out_of_place_only, out_of_place_only=args.out_of_place_only,

2
torch/_inductor/codegen/common.py
View File

@ -232,7 +232,7 @@ def get_wrapper_codegen_for_device(device: str, cpp_wrapper: bool = False):
def init_backend_registration(): def init_backend_registration():
from .cpp import CppScheduling from .cpp import CppScheduling
from .cpp_wrapper_cpu import CppWrapperCpu from .cpp_wrapper_cpu import CppWrapperCpu
from .cpp_wrapper_cuda import CppWrapperGpu from .cpp_wrapper_gpu import CppWrapperGpu
from .cuda_combined_scheduling import CUDACombinedScheduling from .cuda_combined_scheduling import CUDACombinedScheduling
from .halide import HalideScheduling from .halide import HalideScheduling
from .triton import TritonScheduling from .triton import TritonScheduling

0
torch/_inductor/codegen/cpp_wrapper_cuda.py → torch/_inductor/codegen/cpp_wrapper_gpu.py
View File