Revert "return single op (#71)" (#72)

makslevental · web-flow · commit 874356728534 · 2024-02-23T18:57:50.000Z
diff --git a/mlir/extras/util.py b/mlir/extras/util.py
@@ -118,7 +118,7 @@ def find(op):
                     find(o)
 
     find(op)
-    if len(matching) == 1:
+    if single:
         matching = matching[0]
     return matching
 
