Check cluster_dims attribute exists before access

2025-11-14 06:07:55 +08:00 · 2025-11-05 21:08:08 -08:00
1 changed files with 2 additions and 0 deletions
--- a/torch/_inductor/runtime/triton_heuristics.py
+++ b/torch/_inductor/runtime/triton_heuristics.py
@ -1851,6 +1851,8 @@ class TritonCompileResult(CompileResult[CompiledKernel]):
                else (
                    (binary.metadata.num_ctas, *binary.metadata.cluster_dims)
                    if hasattr(binary, "metadata")
+                    and hasattr(binary.metadata, "num_ctas")
+                    and hasattr(binary.metadata, "cluster_dims")
                    else ()
                )
            ),