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 index 2aae3d9c8ca0..62efa7f64575 100644 --- 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)