pytorch
diff --git a/‎torch/_inductor/codegen/triton_utils.py
Lines changed: 2 additions & 2 deletions b/‎torch/_inductor/codegen/triton_utils.py
Lines changed: 2 additions & 2 deletions
diff --git a/‎torch/_inductor/utils.py
Lines changed: 36 additions & 3 deletions b/‎torch/_inductor/utils.py
Lines changed: 36 additions & 3 deletions
@@ -1,4 +1,4 @@
-from typing import Dict, List
+from typing import Any, Dict, List
 
 import torch
 
@@ -59,7 +59,7 @@ def signature_to_meta(
     }
 
 
-def config_of(args: List[KernelArgType]) -> instance_descriptor:
+def config_of(args: List[KernelArgType]) -> Any:
     def is_aligned(x: KernelArgType, alignment: int, include_tensor: bool) -> bool:
         """
         Roughly follow triton code here:
 
@@ -20,6 +20,7 @@
 import textwrap
 import time
 import unittest
+from dataclasses import fields
 from datetime import datetime
 from io import StringIO
 from typing import (
@@ -634,11 +635,43 @@ def has_incompatible_cudagraph_ops(gm):
     return False
 
 
+# Attempt to import AttrsDescriptor from Triton
 try:
-    from triton.compiler.compiler import AttrsDescriptor as instance_descriptor
+    from triton.compiler.compiler import AttrsDescriptor
+
+    attrs_descriptor_available = True
+    # Determine if 'ids_of_folded_args' is a valid field for AttrsDescriptor
+    ids_of_folded_args_available = "ids_of_folded_args" in [
+        f.name for f in fields(AttrsDescriptor)
+    ]
 except ImportError:
-    # To support older version of triton which does not have AttrsDescriptor
-    # class
+    attrs_descriptor_available = False
+
+# Define `instance_descriptor` function with clear conditional handling
+if attrs_descriptor_available:
+
+    def instance_descriptor(
+        divisible_by_16=None,
+        equal_to_1=None,
+        ids_of_folded_args=None,
+        divisible_by_8=None,
+    ):
+        # Prepare the arguments for AttrsDescriptor
+        kwargs = {
+            "divisible_by_16": divisible_by_16,
+            "equal_to_1": equal_to_1,
+            "divisible_by_8": divisible_by_8,
+        }
+
+        # Conditionally add 'ids_of_folded_args' if it's available in AttrsDescriptor
+        if ids_of_folded_args_available:
+            kwargs["ids_of_folded_args"] = ids_of_folded_args
+
+        # Instantiate AttrsDescriptor with the prepared arguments
+        return AttrsDescriptor(**kwargs)
+
+else:
+    # Define a namedtuple as a fallback when AttrsDescriptor is not available
     instance_descriptor = collections.namedtuple(  # type: ignore[no-redef]
         "instance_descriptor",
         ["divisible_by_16", "equal_to_1", "ids_of_folded_args", "divisible_by_8"],