def __compute_apply_placement(
self,
cpu,
requested_cus,
curr_placement_vectors_static,
predicted_usage_static,
workloads,
ordered_workload_ids_static):
predicted_usage_static_vector = None
if len(predicted_usage_static) > 0:
predicted_usage_static_vector = [predicted_usage_static[w_id] for w_id in ordered_workload_ids_static]
new_placement_vectors = self.__compute_new_placement(
cpu,
requested_cus,
curr_placement_vectors_static,
predicted_usage_static_vector)
tid_2order = cpu.get_natural_indexing_2_original_indexing()
thread_id2workload_ids = defaultdict(list)
for w_ind, v in enumerate(new_placement_vectors):
for i, e in enumerate(v):
if e == 1:
thread_id2workload_ids[tid_2order[i]].append(ordered_workload_ids_static[w_ind])
release_all_threads(cpu, workloads.values())
self.__assign_new_mapping(cpu, thread_id2workload_ids)
# TODO: log what's in print_statistics of compute_v2
return cpu
def assign_threads(self, request: AllocateThreadsRequest) -> AllocateResponse:
cpu = request.get_cpu()
workloads = request.get_workloads()
workload_id = request.get_workload_id()
burst_workloads = get_burst_workloads(workloads.values())
release_all_threads(cpu, burst_workloads)
if workloads[workload_id].get_type() == STATIC:
self.__assign_threads(cpu, workloads[workload_id])
update_burst_workloads(cpu, workloads, self.__free_thread_provider)
return AllocateResponse(cpu, self.get_name())
def free_threads(self, request: AllocateThreadsRequest) -> AllocateResponse:
cpu = request.get_cpu()
workloads = request.get_workloads()
workload_id = request.get_workload_id()
burst_workloads = get_burst_workloads(workloads.values())
release_all_threads(cpu, burst_workloads)
for t in cpu.get_threads():
if workload_id in t.get_workload_ids():
t.free(workload_id)
workloads.pop(workload_id)
update_burst_workloads(cpu, workloads, self.__free_thread_provider)
return AllocateResponse(cpu, self.get_name())
def free_threads(self,
request: AllocateThreadsRequest) -> AllocateResponse:
cpu = request.get_cpu()
workloads = request.get_workloads()
workload_id = request.get_workload_id()
burst_workloads = get_burst_workloads(workloads.values())
release_all_threads(cpu, burst_workloads)
if workloads[workload_id].get_type() == STATIC:
self.__free_threads(cpu, workload_id, workloads)
workloads.pop(workload_id)
metadata = {}
update_burst_workloads(cpu, workloads, self.__free_thread_provider,
metadata)
return AllocateResponse(
cpu, get_workload_allocations(cpu, workloads.values()),
self.get_name(), metadata)