[inductor] Make load_mask() codegen determinstic (#126017)

Pull Request resolved: #126017 Approved by: https://github.com/shunting314
pytorch · May 13, 2024 · 6370749 · 6370749
1 parent 82edc8b
commit 6370749
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torch/_inductor/codegen/triton.py b/torch/_inductor/codegen/triton.py
@@ -1844,7 +1844,8 @@ def load_mask(self, var):
             mask = (
                 f"{next(iter(mask_vars))}"
                 if len(mask_vars) == 1
-                else f"({' & '.join(str(v) for v in mask_vars)})"
+                # sorted for deterministic order
+                else f"({' & '.join(sorted(map(str, mask_vars)))})"
             )
         return mask