From edf266e9bbbf6063f7c4a336ffb50234e11a0a82 Mon Sep 17 00:00:00 2001 From: Gabriel Ferns Date: Tue, 28 Jan 2025 20:22:20 +0000 Subject: [PATCH] inductor.config.descriptive_names = False is not actually supported (#145523) Summary: This config is not supported (it throws an error when set), and doesn't really make sense imo. Pull Request resolved: https://github.com/pytorch/pytorch/pull/145523 Approved by: https://github.com/eellison --- torch/_inductor/config.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/torch/_inductor/config.py b/torch/_inductor/config.py index 037a1d68bb1..1eed29c9a11 100644 --- a/torch/_inductor/config.py +++ b/torch/_inductor/config.py @@ -874,7 +874,7 @@ class cpp: # similar to config.triton.descriptive_names descriptive_names: Union[ - bool, Literal["torch", "original_aten", "inductor_node"] + Literal["torch", "original_aten", "inductor_node"] ] = "original_aten" # how many nodes to allow into a single horizontal fusion @@ -1031,12 +1031,11 @@ 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[ - bool, Literal["torch", "original_aten", "inductor_node"] + Literal["torch", "original_aten", "inductor_node"] ] = "original_aten" # use alternate codegen for smaller reductions