diff --git ql/src/java/org/apache/hadoop/hive/ql/exec/tez/DagUtils.java ql/src/java/org/apache/hadoop/hive/ql/exec/tez/DagUtils.java index 78e965b..c0755f8 100644 --- ql/src/java/org/apache/hadoop/hive/ql/exec/tez/DagUtils.java +++ ql/src/java/org/apache/hadoop/hive/ql/exec/tez/DagUtils.java @@ -115,6 +115,7 @@ import org.apache.tez.runtime.library.input.ShuffledUnorderedKVInput; import org.apache.tez.runtime.library.output.OnFileSortedOutput; import org.apache.tez.runtime.library.output.OnFileUnorderedKVOutput; +import org.apache.tez.runtime.library.output.OnFileUnorderedPartitionedKVOutput; /** * DagUtils. DagUtils is a collection of helper methods to convert @@ -311,7 +312,7 @@ private EdgeProperty createEdgeProperty(TezEdgeProperty edgeProp) throws IOExcep case CUSTOM_EDGE: dataMovementType = DataMovementType.CUSTOM; - logicalOutputClass = OnFileSortedOutput.class; + logicalOutputClass = OnFileUnorderedPartitionedKVOutput.class; logicalInputClass = ShuffledUnorderedKVInput.class; EdgeManagerDescriptor edgeDesc = new EdgeManagerDescriptor( CustomPartitionEdge.class.getName()); @@ -331,7 +332,7 @@ private EdgeProperty createEdgeProperty(TezEdgeProperty edgeProp) throws IOExcep case CUSTOM_SIMPLE_EDGE: dataMovementType = DataMovementType.SCATTER_GATHER; - logicalOutputClass = OnFileSortedOutput.class; + logicalOutputClass = OnFileUnorderedPartitionedKVOutput.class; logicalInputClass = ShuffledUnorderedKVInput.class; break;