Fix failed to get iterator exception

apache · Nov 7, 2023 · 8ac788c · 8ac788c
1 parent ba045c7
commit 8ac788c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/backends-velox/src/main/scala/io/glutenproject/backendsapi/velox/IteratorApiImpl.scala b/backends-velox/src/main/scala/io/glutenproject/backendsapi/velox/IteratorApiImpl.scala
@@ -149,6 +149,7 @@ class IteratorApiImpl extends IteratorApi with Logging {
 
     Iterators
       .wrap(resIter.asScala)
+      .protectInvocationFlow()
       .recycleIterator {
         updateNativeMetrics(resIter.getMetrics)
         updateInputMetrics(TaskContext.get().taskMetrics().inputMetrics)