OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Hipify global scrach defintion in AOTI codegen (#150893)

Browse Source 
Summary: as title, a refactor is very needed I think .... or at least unify internal/external AOTI wrapper hipification method

Test Plan: P1780296121

Differential Revision: D72683568

Pull Request resolved: https://github.com/pytorch/pytorch/pull/150893
Approved by: https://github.com/davidberard98
This commit is contained in:
Zhuoran Zhao 2025-04-09 18:35:36 +00:00 committed by PyTorch MergeBot
parent d04a6ec021
commit d3a2872c67
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -440,7 +440,7 @@ class CppWrapperGpu(CppWrapperCpu):
is not None
):
global_scratch_def, global_scratch_var = global_scratch
code.writeline(global_scratch_def)
code.writeline(maybe_hipify_code_wrapper(global_scratch_def))
new_args.append(f"&{global_scratch_var}")
return ", ".join(new_args)