Skip to content

Commit d833f49

Browse files
desertfirepytorchmergebot
authored andcommitted
[reland][Inductor] Rename cpp_wrapper_cuda.py as cpp_wrapper_gpu.py (pytorch#136046)
Summary: Reland pytorch#135313 after fixing internal build issues Test Plan: CI Differential Revision: D62658837 Pull Request resolved: pytorch#136046 Approved by: https://github.com/chenyang78, https://github.com/etaf, https://github.com/jansel
1 parent a803cb0 commit d833f49

File tree

3 files changed

+2
-2
lines changed

3 files changed

+2
-2
lines changed

tools/amd_build/build_amd.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -207,7 +207,7 @@ def remove_hcc(line: str) -> str:
207207
ignores=ignores,
208208
extra_files=[
209209
"torch/_inductor/codegen/cpp_wrapper_cpu.py",
210-
"torch/_inductor/codegen/cpp_wrapper_cuda.py",
210+
"torch/_inductor/codegen/cpp_wrapper_gpu.py",
211211
"torch/_inductor/codegen/wrapper.py",
212212
],
213213
out_of_place_only=args.out_of_place_only,

torch/_inductor/codegen/common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -232,7 +232,7 @@ def get_wrapper_codegen_for_device(device: str, cpp_wrapper: bool = False):
232232
def init_backend_registration():
233233
from .cpp import CppScheduling
234234
from .cpp_wrapper_cpu import CppWrapperCpu
235-
from .cpp_wrapper_cuda import CppWrapperGpu
235+
from .cpp_wrapper_gpu import CppWrapperGpu
236236
from .cuda_combined_scheduling import CUDACombinedScheduling
237237
from .halide import HalideScheduling
238238
from .triton import TritonScheduling

0 commit comments

Comments
 (0)