Python model_from_valuesの例

コード例 #1

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

コード例 #2

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