From 1a297b58bf40bcb92ab74f417fb02f0360850ecc Mon Sep 17 00:00:00 2001 From: Jason Ansel Date: Sat, 11 May 2024 21:01:43 -0700 Subject: [PATCH 1/2] Update [ghstack-poisoned] --- 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 81fc19a0f3f46..c8f0deca4b965 100644 --- a/torch/_inductor/codegen/triton.py +++ b/torch/_inductor/codegen/triton.py @@ -1842,7 +1842,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(str(v) for v in sorted(mask_vars, key=str))})" ) return mask From 273211f27976baedb82274c47bc019242f8352fd Mon Sep 17 00:00:00 2001 From: Jason Ansel Date: Sun, 12 May 2024 08:50:04 -0700 Subject: [PATCH 2/2] Update [ghstack-poisoned] --- torch/_inductor/codegen/triton.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch/_inductor/codegen/triton.py b/torch/_inductor/codegen/triton.py index 42f41881f0f74..5ad5f791a9023 100644 --- a/torch/_inductor/codegen/triton.py +++ b/torch/_inductor/codegen/triton.py @@ -1845,7 +1845,7 @@ def load_mask(self, var): f"{next(iter(mask_vars))}" if len(mask_vars) == 1 # sorted for deterministic order - else f"({' & '.join(str(v) for v in sorted(mask_vars, key=str))})" + else f"({' & '.join(sorted(map(str, mask_vars)))})" ) return mask