pavel/fedavg_mpi
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

commit small improvements before going craZy

Browse Source
This commit is contained in:
Pavel Lutskov
2019-11-25 20:54:27 -08:00
parent 04e0b9829c
commit 253c833b5b
2 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
main.c
View File

@@ -92,14 +92,11 @@ void slave_node() {
MPI_Recv(shape, 2, MPI_LONG, P_READER, MPI_ANY_TAG, MPI_COMM_WORLD,
MPI_STATUS_IGNORE);
long size = shape[0] * shape[1];
float* data = malloc(shape[0] * shape[1] * sizeof(float));
MPI_Recv(data, size, MPI_FLOAT, P_READER, MPI_ANY_TAG,
float* batch = malloc(shape[0] * shape[1] * sizeof(float));
MPI_Recv(batch, size, MPI_FLOAT, P_READER, MPI_ANY_TAG,
MPI_COMM_WORLD, MPI_STATUS_IGNORE);
PyArrayObject* batch = PyArray_SimpleNewFromData(
2, shape, NPY_FLOAT32, data);
step_net(net, batch);
Py_DECREF(batch);
free(data);
step_net(net, batch, BS);
free(batch);
}
Network c_net;
cify_network(net, &c_net);