Update test/test_meta.py

pytorch · Dec 7, 2022 · 6519637 · 6519637
1 parent 281552a
commit 6519637
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_meta.py b/test/test_meta.py
@@ -364,7 +364,7 @@ def test_assert(cond, msg):
         if should_check_strides(func) == CheckStrides.ALL:
             same_strides, _ = torch._prims_common.check_all_strides(meta_r, r)
             test_assert(same_strides, f"but real stride was {r.stride()}")
-        if should_check_strides(func) == CheckStrides.SIGNIFICANT:
+        elif should_check_strides(func) == CheckStrides.SIGNIFICANT:
             same_strides, _ = torch._prims_common.check_significant_strides(meta_r, r)
             test_assert(same_strides, f"but real stride was {r.stride()}")
         test_assert(