Exemple #1
0
def commtime_dict_togpu(network, nbytes=[10, 100, 1000, 10000]):
    return {(host, gpu):
                merge(network[host, gpu],
                      model_from_values(comm_times_togpu(nbytes, host)))
            for host, gpu in network
            if network[host, gpu]['type'] == 'togpu'}
Exemple #2
0
def commtime_dict_fromgpu(network, nbytes=[10, 100, 1000, 10000]):
    return {(gpu, host):
            merge(network[gpu, host],
                  model_from_values(comm_times_fromgpu(nbytes, host)))
            for gpu, host in network
            if network[gpu, host]['type'] == 'fromgpu'}