microsoft
/
CNTK
зеркало из
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность

Debug:TestNet

This commit is contained in:
Qiwei Ye 2016-03-02 00:37:59 +08:00
Родитель 39f1d7628f
Коммит a335f2c4d1
1 изменённых файлов: 34 добавлений и 30 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

30
Source/Common/Include/MultiversoWrapper.h
Просмотреть файл

@ -392,54 +392,58 @@ namespace Microsoft {
strcpy(hi2, chi2); strcpy(hi2, chi2);
char* hi3 = new char[19]; char* hi3 = new char[19];
strcpy(hi3, chi3); strcpy(hi3, chi3);
//char* hi2 = "hello, c++";
//char* hi3 = "hello, multiverso";
if (net->rank() == 0) { if (net->rank() == 0) {
for (int rank = 1; rank < net->size(); ++rank) {
multiverso::MessagePtr msg(new multiverso::Message());// = std::make_unique<Message>(); multiverso::MessagePtr msg(new multiverso::Message());// = std::make_unique<Message>();
msg->set_src(0); msg->set_src(0);
msg->set_dst(1); msg->set_dst(rank);
msg->Push(multiverso::Blob(hi1, 13)); msg->Push(multiverso::Blob(hi1, 13));
msg->Push(multiverso::Blob(hi2, 11)); msg->Push(multiverso::Blob(hi2, 11));
msg->Push(multiverso::Blob(hi3, 18)); msg->Push(multiverso::Blob(hi3, 18));
net->Send(msg); net->Send(msg);
multiverso::Log::Info("rank 0 send\n"); multiverso::Log::Info("rank 0 send\n");
multiverso::Log::Info("Hi = %s\n", msg->data()[0].data()); }
for (int i = 1; i < net->size(); ++i) {
multiverso::MessagePtr msg(new multiverso::Message());
msg.reset(new multiverso::Message()); msg.reset(new multiverso::Message());
while (net->Recv(&msg) == 0) { while (net->Recv(&msg) == 0) {
multiverso::Log::Info("recv return 0\n"); // multiverso::Log::Info("recv return 0\n");
} }
multiverso::Log::Info("rank 0 recv\n"); multiverso::Log::Info("rank 0 recv\n");
// CHECK(strcmp(msg->data()[0].data(), hi) == 0); // CHECK(strcmp(msg->data()[0].data(), hi) == 0);
std::vector<multiverso::Blob> recv_data = msg->data(); std::vector<multiverso::Blob> recv_data = msg->data();
//CHECK(recv_data.size() == 3); //CHECK(recv_data.size() == 3);
for (int i = 0; i < msg->size(); ++i) { for (int i = 0; i < msg->size(); ++i) {
multiverso::Log::Info("%s\n", recv_data[i].data()); multiverso::Log::Info("recv from srv %d: %s\n", msg->src(), recv_data[i].data());
}; };
} }
else if (net->rank() == 1) { }
else {// other rank
multiverso::MessagePtr msg(new multiverso::Message());// = std::make_unique<Message>(); multiverso::MessagePtr msg(new multiverso::Message());// = std::make_unique<Message>();
while (net->Recv(&msg) == 0) { while (net->Recv(&msg) == 0) {
multiverso::Log::Info("recv return 0\n"); // multiverso::Log::Info("recv return 0\n");
} }
multiverso::Log::Info("rank 1 recv\n"); multiverso::Log::Info("rank %d recv\n", net->rank());
// CHECK(strcmp(msg->data()[0].data(), hi) == 0); // CHECK(strcmp(msg->data()[0].data(), hi) == 0);
std::vector<multiverso::Blob> recv_data = msg->data(); std::vector<multiverso::Blob>& recv_data = msg->data();
//CHECK(recv_data.size() == 3); //CHECK(recv_data.size() == 3);
for (int i = 0; i < msg->size(); ++i) { for (int i = 0; i < msg->size(); ++i) {
multiverso::Log::Info("%s\n", recv_data[i].data()); multiverso::Log::Info("%s\n", recv_data[i].data());
} }
msg.reset(new multiverso::Message()); msg.reset(new multiverso::Message());
msg->set_src(1); msg->set_src(net->rank());
msg->set_dst(0); msg->set_dst(0);
msg->Push(multiverso::Blob(hi1, 13)); msg->Push(multiverso::Blob(hi1, 13));
msg->Push(multiverso::Blob(hi2, 11)); msg->Push(multiverso::Blob(hi2, 11));
msg->Push(multiverso::Blob(hi3, 18)); msg->Push(multiverso::Blob(hi3, 18));
net->Send(msg); net->Send(msg);
multiverso::Log::Info("rank 0 send\n"); multiverso::Log::Info("rank %d send\n", net->rank());
multiverso::Log::Info("Hi = %s\n", msg->data()[0].data());
} }
// while (!net->Test()) {
// // wait all message process finished
// }
net->Finalize(); net->Finalize();
} }