diff --git ql/src/java/org/apache/hadoop/hive/ql/parse/GenTezUtils.java ql/src/java/org/apache/hadoop/hive/ql/parse/GenTezUtils.java index 32bf24d..43b534a 100644 --- ql/src/java/org/apache/hadoop/hive/ql/parse/GenTezUtils.java +++ ql/src/java/org/apache/hadoop/hive/ql/parse/GenTezUtils.java @@ -140,12 +140,14 @@ public static ReduceWork createReduceWork( TezEdgeProperty edgeProp; EdgeType edgeType = determineEdgeType(context.preceedingWork, reduceWork); - if (reduceWork.isAutoReduceParallelism()) { + if (reduceWork.isAutoReduceParallelism() + && edgeType != EdgeType.CUSTOM_SIMPLE_EDGE) { edgeProp = new TezEdgeProperty(context.conf, edgeType, true, reduceWork.getMinReduceTasks(), reduceWork.getMaxReduceTasks(), bytesPerReducer); } else { edgeProp = new TezEdgeProperty(edgeType); + reduceWork.setAutoReduceParallelism(false); } tezWork.connect(