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] Make the original ATen info dumped in alphabetical order (#97261)

Browse Source 
Summary: To avoid a lot of noises when comparing output_code.py from two
runs.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/97261
Approved by: https://github.com/Chillee
This commit is contained in:
Bin Bao
2023-03-21 16:51:51 +00:00
committed by PyTorch MergeBot
parent 01885cea43
commit ea9194a4f2
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
torch/_inductor/utils.py
View File

@ -283,15 +283,13 @@ def get_kernel_metadata(node_schedule):
for node in inductor_nodes:
if "original_aten" in node.meta:
original_aten_dict[str(node.meta["original_aten"]._overloadpacket)].append(
node
node.name
)
metadata = [
f"# Original ATen: {', '.join(original_aten_dict.keys())}\n",
f"# Original ATen: {', '.join(sorted(original_aten_dict.keys()))}\n",
]
for original_aten, nodes in original_aten_dict.items():
metadata.append(
f"# {original_aten} => {', '.join([node.name for node in nodes])}"
)
for original_aten, nodes in sorted(original_aten_dict.items()):
metadata.append(f"# {original_aten} => {', '.join(sorted(nodes))}")
return "\n".join(metadata)