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'}
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'}