Wrong Split computation causing less number of container launch

Author: Aggarwal-Raghav

ql/src/java/org/apache/hadoop/hive/ql/exec/tez/HiveSplitGenerator.java

@@ -359,8 +359,11 @@ public List<Event> initialize() throws Exception {
             splits[0] = new HiveInputFormat.HiveInputSplit(fileSplit, partIF);
           }
         } else {
-          // Raw splits
-          splits = inputFormat.getSplits(jobConf, numSplits.orElse(Math.multiplyExact(availableSlots, (int)waves)));
+          // Raw splits.
+          splits =
+              inputFormat.getSplits(
+                  jobConf,
+                  numSplits.orElse((int) Math.min(availableSlots * waves, Integer.MAX_VALUE)));
         }
         // Sort the splits, so that subsequent grouping is consistent.
         Arrays.sort(splits, new InputSplitComparator());

ql/src/test/results/clientpositive/llap/mm_all.q.out

@@ -1998,7 +1998,7 @@ Table Type:         	MANAGED_TABLE
 Table Parameters:	 	 
 	COLUMN_STATS_ACCURATE	{\"BASIC_STATS\":\"true\"}
 	bucketing_version   	2                   
-	numFiles            	16                  
+	numFiles            	27                  
 	numRows             	500                 
 	rawDataSize         	5312                
 	totalSize           	#Masked#