diff --git a/ql/src/java/org/apache/hadoop/hive/ql/exec/tez/tools/KeyValueInputMerger.java b/ql/src/java/org/apache/hadoop/hive/ql/exec/tez/tools/KeyValueInputMerger.java index f5591d1bb0..8b3abb971b 100644 --- a/ql/src/java/org/apache/hadoop/hive/ql/exec/tez/tools/KeyValueInputMerger.java +++ b/ql/src/java/org/apache/hadoop/hive/ql/exec/tez/tools/KeyValueInputMerger.java @@ -91,10 +91,12 @@ public KeyValueInputMerger(List multiMRInputs, Deserializer dese List splits = split.getGroupedSplits(); // There maybe more than 1 splits in the group, however, they all have 1 unique path. // Assert that. - Path path = ((HiveInputFormat.HiveInputSplit) splits.get(0)).getPath(); - Path pathFromMap = kvReaderPathMap.putIfAbsent(input, path); - if (pathFromMap != null) { - assert pathFromMap.equals(path); + for (int i = 0; i < splits.size(); i++) { + Path path = ((HiveInputFormat.HiveInputSplit) splits.get(i)).getPath(); + Path pathFromMap = kvReaderPathMap.putIfAbsent(input, path); + if (pathFromMap != null) { + assert pathFromMap.equals(path); + } } addToQueue(input); }