[Inductor] Support Triton AttrsDescriptor cls field (pytorch#139193)

Fixes pytorch#139179 Adding corresponding changes to triton-lang/triton#4888 Pull Request resolved: pytorch#139193 Approved by: https://github.com/bertmaher
Ryo-not-rio · Oct 30, 2024 · 5c6d354 · 5c6d354
1 parent 180d283
commit 5c6d354
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/torch/_inductor/runtime/hints.py b/torch/_inductor/runtime/hints.py
@@ -52,7 +52,9 @@ def AttrsDescriptorWrapper(
             }
 
             # Instantiate AttrsDescriptor with the prepared arguments
-            res = AttrsDescriptor.from_dict(kwargs)
+            res = AttrsDescriptor.from_dict(
+                {"arg_properties": kwargs, "cls": AttrsDescriptor.__name__}
+            )
             assert res.property_values["tt.divisibility"] == 16
             assert res.property_values["tt.equal_to"] == 1
             return res