return single op (#71)

makslevental · web-flow · commit e3a880d7075b · 2024-02-23T17:15:21.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 single:
+    if len(matching) == 1:
         matching = matching[0]
     return matching
 
