diff --git ql/src/java/org/apache/hadoop/hive/ql/exec/vector/expressions/aggregates/VectorUDAFCount.java ql/src/java/org/apache/hadoop/hive/ql/exec/vector/expressions/aggregates/VectorUDAFCount.java index 4e43905..cf373a1 100644 --- ql/src/java/org/apache/hadoop/hive/ql/exec/vector/expressions/aggregates/VectorUDAFCount.java +++ ql/src/java/org/apache/hadoop/hive/ql/exec/vector/expressions/aggregates/VectorUDAFCount.java @@ -97,7 +97,12 @@ public void aggregateInputSelection( ColumnVector inputVector = batch.cols[this.inputExpression.getOutputColumn()]; - if (inputVector.noNulls) { + if (inputVector.isRepeating) { + if (inputVector.noNulls || !inputVector.isNull[0]) { + iterateNoNullsWithAggregationSelection( + aggregationBufferSets, aggregateIndex, batchSize); + } + } else if (inputVector.noNulls) { // if there are no nulls then the iteration is the same on all cases iterateNoNullsWithAggregationSelection( aggregationBufferSets, aggregateIndex, batchSize);