diff --git a/ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/Vectorizer.java b/ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/Vectorizer.java index f4b4c2ff3bad..08d396ae3f4c 100644 --- a/ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/Vectorizer.java +++ b/ql/src/java/org/apache/hadoop/hive/ql/optimizer/physical/Vectorizer.java @@ -5161,13 +5161,11 @@ private static VectorPTFInfo createVectorPTFInfo(Operator partitionExpressionByOutputColumn = new HashMap<>(); for (int i = 0; i < partitionKeyCount; i++) { VectorExpression partitionExpression = vContext.getVectorExpression(partitionExprNodeDescs[i]); - TypeInfo typeInfo = partitionExpression.getOutputTypeInfo(); - Type columnVectorType = VectorizationContext.getColumnVectorTypeFromTypeInfo(typeInfo); - partitionColumnVectorTypes[i] = columnVectorType; partitionColumnMap[i] = partitionExpression.getOutputColumnNum(); - partitionExpressions[i] = partitionExpression; + partitionExpressionByOutputColumn.put(partitionColumnMap[i], partitionExpression); } final int orderKeyCount = orderExprNodeDescs.length; @@ -5190,6 +5188,23 @@ private static VectorPTFInfo createVectorPTFInfo(Operator