diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuColumnarToRowExec.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuColumnarToRowExec.scala index 1d1e21873b2..3435de0891f 100644 --- a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuColumnarToRowExec.scala +++ b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuColumnarToRowExec.scala @@ -281,16 +281,16 @@ abstract class GpuColumnarToRowExecParent(child: SparkPlan, val exportColumnarRd override lazy val allMetrics: Map[String, GpuMetric] = Map( NUM_OUTPUT_ROWS -> createMetric(outputRowsLevel, DESCRIPTION_NUM_OUTPUT_ROWS), OP_TIME -> createNanoTimingMetric(MODERATE_LEVEL, DESCRIPTION_OP_TIME), - FETCH_TIME -> createNanoTimingMetric(MODERATE_LEVEL, DESCRIPTION_FETCH_TIME), + COLLECT_TIME -> createNanoTimingMetric(MODERATE_LEVEL, DESCRIPTION_COLLECT_TIME), NUM_INPUT_BATCHES -> createMetric(DEBUG_LEVEL, DESCRIPTION_NUM_INPUT_BATCHES)) override def doExecute(): RDD[InternalRow] = { val numOutputRows = gpuLongMetric(NUM_OUTPUT_ROWS) val numInputBatches = gpuLongMetric(NUM_INPUT_BATCHES) val opTime = gpuLongMetric(OP_TIME) - val fetchTime = gpuLongMetric(FETCH_TIME) + val collectTime = gpuLongMetric(COLLECT_TIME) - val f = makeIteratorFunc(child.output, numOutputRows, numInputBatches, opTime, fetchTime) + val f = makeIteratorFunc(child.output, numOutputRows, numInputBatches, opTime, collectTime) val cdata = child.executeColumnar() if (exportColumnarRdd) { diff --git a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuExec.scala b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuExec.scala index dd62f1e1161..b04f32865a1 100644 --- a/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuExec.scala +++ b/sql-plugin/src/main/scala/com/nvidia/spark/rapids/GpuExec.scala @@ -49,7 +49,6 @@ object GpuMetric extends Logging { val NUM_PARTITIONS = "numPartitions" val TOTAL_TIME = "totalTime" val OP_TIME = "opTime" - val FETCH_TIME = "fetchTime" val PEAK_DEVICE_MEMORY = "peakDevMemory" val COLLECT_TIME = "collectTime" val CONCAT_TIME = "concatTime" @@ -76,7 +75,6 @@ object GpuMetric extends Logging { val DESCRIPTION_NUM_PARTITIONS = "partitions" val DESCRIPTION_TOTAL_TIME = "total time" val DESCRIPTION_OP_TIME = "op time" - val DESCRIPTION_FETCH_TIME = "fetch time" val DESCRIPTION_PEAK_DEVICE_MEMORY = "peak device memory" val DESCRIPTION_COLLECT_TIME = "collect batch time" val DESCRIPTION_CONCAT_TIME = "concat batch time"