From 637074983e1f08abcf8a874001d5d6a7e637ace7 Mon Sep 17 00:00:00 2001 From: Jason Ansel Date: Sun, 12 May 2024 08:50:04 -0700 Subject: [PATCH] [inductor] Make load_mask() codegen determinstic (#126017) Pull Request resolved: https://github.com/pytorch/pytorch/pull/126017 Approved by: https://github.com/shunting314 --- torch/_inductor/codegen/triton.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/torch/_inductor/codegen/triton.py b/torch/_inductor/codegen/triton.py index 68c3cb86e72b6..5ad5f791a9023 100644 --- 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