Added code to bump resources (#94)

data/alibaba_loader.py

@@ -114,6 +114,7 @@ def __init__(
  self._task_cpu_usage_min = self._flags.alibaba_loader_task_cpu_usage_min
  self._task_cpu_usage_max = self._flags.alibaba_loader_task_cpu_usage_max
  self._task_cpu_usage_random = self._flags.alibaba_loader_task_cpu_usage_random
+ self._alibaba_bump_resources_of_low_duration_task = self._flags.alibaba_bump_resources_of_low_duration_task
 
  def _construct_workload_definitions(
  self,
@@ -549,6 +550,10 @@ def _convert_job_data_to_job_graph(
  )
  * self._task_cpu_multiplier
  )
+ # bump up the resources of low duration tasks
+ if self._alibaba_bump_resources_of_low_duration_task and task.actual_duration < 4:
+ resource_usage = self._task_cpu_usage_max
+
  job_resources_1 = Resources(
  resource_vector={
  Resource(name="Slot_1", _id="any"): resource_usage,

main.py

@@ -160,6 +160,11 @@
  "The divisor used when converting alibaba trace tasks task.cpu to slot. The lower "
  "the number, the higher the slots.",
 )
+flags.DEFINE_bool(
+ "alibaba_bump_resources_of_low_duration_task",
+ False,
+ "If True, the lower duration task resources would be bumped to max",
+)
 flags.DEFINE_bool(
  "alibaba_loader_task_cpu_usage_random",
  False,