def large_job_pruning()

in cluster-trace-gpu-v2020/simulator/utils.py [0:0]

• 13 lines of code
• 7 McCabe index (conditional complexity)

def large_job_pruning(job_list, gpu_limit, cpu_limit):
    if job_list is None:
        return []
    for job in job_list:
        if 'num_gpu' in job and job['num_gpu'] > gpu_limit:
            gpu_was = job['num_gpu']
            job['num_gpu'] = gpu_limit
            print_fn("{:s}: GPU {:d} ==> {:d}".format(_repr_job_concise(job), gpu_was, gpu_limit))
        if 'num_cpu' in job and job['num_cpu'] > cpu_limit:
            cpu_was = job['num_cpu']
            job['num_cpu'] = cpu_limit
            print_fn("{:s}: CPU {:d} ==> {:d}".format(_repr_job_concise(job), cpu_was, cpu_limit))
    return job_list