diff --git ql/src/java/org/apache/hadoop/hive/ql/exec/tez/CustomPartitionVertex.java ql/src/java/org/apache/hadoop/hive/ql/exec/tez/CustomPartitionVertex.java index cecedfb..4bc50a4 100644 --- ql/src/java/org/apache/hadoop/hive/ql/exec/tez/CustomPartitionVertex.java +++ ql/src/java/org/apache/hadoop/hive/ql/exec/tez/CustomPartitionVertex.java @@ -90,9 +90,9 @@ public void initialize(VertexManagerPluginContext context) { @Override public void onVertexStarted(Map> completions) { int numTasks = context.getVertexNumTasks(context.getVertexName()); - List scheduledTasks = new ArrayList(numTasks); + List scheduledTasks = new ArrayList(numTasks); for (int i = 0; i < numTasks; ++i) { - scheduledTasks.add(new Integer(i)); + scheduledTasks.add(new VertexManagerPluginContext.TaskWithLocationHint(new Integer(i), null)); } context.scheduleVertexTasks(scheduledTasks); } @@ -195,7 +195,7 @@ public void onRootVertexInitialized(String inputName, InputDescriptor inputDescr getBucketSplitMapForPath(pathFileSplitsMap); try { - int totalResource = context.getTotalAVailableResource().getMemory(); + int totalResource = context.getTotalAvailableResource().getMemory(); int taskResource = context.getVertexTaskResource().getMemory(); float waves = conf.getFloat(TezConfiguration.TEZ_AM_GROUPING_SPLIT_WAVES,