def Comp_with_aiob()

in utils/utils.py [0:0]

• 9 lines of code
• 5 McCabe index (conditional complexity)

def Comp_with_aiob(workload, compute_cache):
    for item in workload.workload:
        if item.comm_type == CommType.computation:
            for key in compute_cache:
                key_temp = key.split("_")[0]
                if key_temp in item.stage:
                    item._elapsed_time = compute_cache[key]
                    break
    return workload