From b60fe754cd8e3459740b5e0423144346245796a4 Mon Sep 17 00:00:00 2001 From: JiaKe Date: Wed, 8 Nov 2023 11:06:14 +0800 Subject: [PATCH] [VL] Fix failed to get iterator exception from file format writer (#3641) --- .../io/glutenproject/backendsapi/velox/IteratorApiImpl.scala | 1 + 1 file changed, 1 insertion(+) 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 36ce0ffae135..114de2b623fd 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 @@ -150,6 +150,7 @@ class IteratorApiImpl extends IteratorApi with Logging { Iterators .wrap(resIter.asScala) + .protectInvocationFlow() .recycleIterator { updateNativeMetrics(resIter.getMetrics) updateInputMetrics(TaskContext.get().taskMetrics().inputMetrics)