default val on function

xdslproject · Dec 20, 2024 · 0e461a8 · 0e461a8
1 parent 1ffe566
commit 0e461a8
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/xdsl/transforms/memref_to_dsd.py b/xdsl/transforms/memref_to_dsd.py
@@ -130,8 +130,7 @@ def match_and_rewrite(self, op: memref.SubviewOp, rewriter: PatternRewriter, /):
             # 1d subview onto a nd memref
             sizes = op.static_sizes.get_values()
             counter_sizes = collections.Counter(sizes)
-            if 1 in counter_sizes:
-                counter_sizes.pop(1)
+            counter_sizes.pop(1, None)
             assert (
                 len(counter_sizes) == 1
             ), "1d access into nd memref must specify one size > 1"