Update

laithsakka · laithsakka · commit c00817f2c6eb · 2025-09-08T10:19:25.000-07:00
[ghstack-poisoned]
diff --git a/torch/_prims_common/__init__.py b/torch/_prims_common/__init__.py
@@ -609,10 +609,10 @@ def should_swap(idx_a, idx_b):
 
             # when stride_a = 1, we want stride_a < stride_b to be TRUE
             # when stride_b = 1, we want stride_a < stride_b to be FALSE
-            if guard_or_false(stride_a == 1):
+            elif guard_or_false(stride_a == 1):
                 return -1
 
-            if guard_or_false(stride_b == 1):
+            elif guard_or_false(stride_b == 1):
                 return 1
 
             if guard_size_oblivious(stride_a < stride_b):
