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

Always produce kernel_info.json (#163715)

Browse Source 
Summary: Always produce kernel_info.json so zoomer can use this json to populate GPU traces

Test Plan: CI

Differential Revision: D82762435

Pull Request resolved: https://github.com/pytorch/pytorch/pull/163715
Approved by: https://github.com/angelayi
This commit is contained in:
Shangdi Yu
2025-09-25 19:38:49 +00:00
committed by PyTorch MergeBot
parent 21a41edd4f
commit ebfc87e303
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
torch/_inductor/codecache.py
View File

@ -2495,16 +2495,16 @@ end
if config.aot_inductor.package:
generated_files.append(output_so)
if config.aot_inductor.package:
if config.trace.provenance_tracking_level != 0:
kernel_info = torch._inductor.debug.create_kernel_information_json()
kernel_info_json = os.path.join(
wrapper_path_operator.parent, "kernel_information.json"
)
with open(kernel_info_json, "w") as f:
f.write(json.dumps(kernel_info, indent=4))
generated_files.append(kernel_info_json)
if config.trace.provenance_tracking_level != 0:
kernel_info = torch._inductor.debug.create_kernel_information_json()
kernel_info_json = os.path.join(
wrapper_path_operator.parent, "kernel_information.json"
)
with open(kernel_info_json, "w") as f:
f.write(json.dumps(kernel_info, indent=4))
generated_files.append(kernel_info_json)
if config.aot_inductor.package:
# We want to return the directory that contains all the AOTI
# generated files, not just the so
# return os.path.split(output_so)[0]