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

Revert "inductor.config.descriptive_names = False is not actually supported (#145523) (#146051) (#151481)"

Browse Source 
This reverts commit cfc4d74b0c.

Reverted https://github.com/pytorch/pytorch/pull/151481 on behalf of https://github.com/malfet due to It indeed breaks lint, it followup PR contains it's own issues ([comment](https://github.com/pytorch/pytorch/pull/151481#issuecomment-2816490764))
This commit is contained in:
PyTorch MergeBot 2025-04-19 03:12:56 +00:00
parent 843e4d11ba
commit 6261db7719
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
torch/_inductor/config.py
View File

@ -947,7 +947,7 @@ class cpp:
# similar to config.triton.descriptive_names
descriptive_names: Union[
Literal["torch", "original_aten", "inductor_node"]
bool, Literal["torch", "original_aten", "inductor_node"]
] = "original_aten"
# how many nodes to allow into a single horizontal fusion
@ -1118,11 +1118,12 @@ class triton:
)
# should we put op names in kernel names
# False: No special names (just triton__1, triton__2, etc.)
# "torch": Maps to the fx op in the Dynamo graph (module name, method name, etc.)
# "original_aten": Maps to the highest-level aten op (i.e. pre-decompositions)
# "inductor_node": Maps to the node name in the FX graph passed to Inductor
descriptive_names: Union[
Literal["torch", "original_aten", "inductor_node"]
bool, Literal["torch", "original_aten", "inductor_node"]
] = "original_aten"
# use alternate codegen for smaller reductions