You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
118 lines
3.3 KiB
118 lines
3.3 KiB
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "libpaddle_pserver_cclient.h"
|
|
|
|
typedef float real;
|
|
|
|
void fail() {
|
|
// TODO(helin): fix: gtest using cmake is not working, using this
|
|
// hacky way for now.
|
|
printf("test failed.\n");
|
|
exit(-1);
|
|
}
|
|
|
|
void print_parameter(paddle_gradient* param) {
|
|
if (param == NULL) {
|
|
printf("param is NULL!!\n");
|
|
} else {
|
|
printf("==== parameter ====\n");
|
|
printf("name: %s\n", param->name);
|
|
printf("content_len: %d\n", param->content_len);
|
|
printf("content_type: %d\n", param->element_type);
|
|
int i;
|
|
for (i = 0; i < param->content_len / (int)sizeof(real); ++i) {
|
|
printf("%f ", ((float*)param->content)[i]);
|
|
}
|
|
printf("\n\n");
|
|
}
|
|
}
|
|
|
|
int main() {
|
|
char addr[] = "localhost:3000";
|
|
paddle_pserver_client c = paddle_new_pserver_client(addr, 1);
|
|
|
|
char* names[] = {"param_a", "param_b"};
|
|
|
|
retry:
|
|
printf("init parameter to pserver:\n");
|
|
|
|
real param_content1[] = {0.1, 0.2, 0.3};
|
|
real param_content2[] = {0.4, 0.5, 0.6};
|
|
paddle_parameter** params =
|
|
(paddle_parameter**)malloc(sizeof(paddle_parameter*) * 2);
|
|
params[0] = (paddle_parameter*)malloc(sizeof(paddle_parameter));
|
|
params[0]->name = names[0];
|
|
params[0]->content = (unsigned char*)param_content1;
|
|
params[0]->content_len = 3 * sizeof(real);
|
|
params[0]->element_type = PADDLE_ELEMENT_TYPE_FLOAT32;
|
|
|
|
params[1] = (paddle_parameter*)malloc(sizeof(paddle_parameter));
|
|
params[1]->name = names[1];
|
|
params[1]->content = (unsigned char*)param_content2;
|
|
params[1]->content_len = 3 * sizeof(real);
|
|
params[1]->element_type = PADDLE_ELEMENT_TYPE_INT32;
|
|
|
|
if (paddle_begin_init_params(c)) {
|
|
if (paddle_init_param(c, *params[0], NULL, 0) != 0) {
|
|
goto retry;
|
|
}
|
|
if (paddle_init_param(c, *params[1], NULL, 0) != 0) {
|
|
goto retry;
|
|
}
|
|
if (paddle_finish_init_params(c) != 0) {
|
|
goto retry;
|
|
}
|
|
} else {
|
|
fail();
|
|
}
|
|
|
|
printf("get inited parameters from pserver:\n");
|
|
// get parameters again by reusing the allocated parameter buffers.
|
|
if (paddle_get_params(c, params, 2) != 0) {
|
|
fail();
|
|
}
|
|
print_parameter(params[0]);
|
|
print_parameter(params[1]);
|
|
|
|
printf("send gradient to pserver:\n");
|
|
real gradient_content1[] = {0.01, 0.02, 0.03};
|
|
real gradinet_content2[] = {0.04, 0.05, 0.06};
|
|
|
|
paddle_gradient** grads =
|
|
(paddle_gradient**)malloc(sizeof(paddle_gradient*) * 2);
|
|
grads[0] = (paddle_gradient*)malloc(sizeof(paddle_gradient));
|
|
grads[0]->name = names[0];
|
|
grads[0]->content = (unsigned char*)gradient_content1;
|
|
grads[0]->content_len = 3 * sizeof(real);
|
|
grads[0]->element_type = PADDLE_ELEMENT_TYPE_FLOAT32;
|
|
|
|
grads[1] = (paddle_gradient*)malloc(sizeof(paddle_gradient));
|
|
grads[1]->name = names[1];
|
|
grads[1]->content = (unsigned char*)gradinet_content2;
|
|
grads[1]->content_len = 3 * sizeof(real);
|
|
grads[1]->element_type = PADDLE_ELEMENT_TYPE_INT32;
|
|
|
|
printf("print gradient sent to pserver:\n");
|
|
print_parameter(grads[0]);
|
|
print_parameter(grads[1]);
|
|
|
|
if (paddle_send_grads(c, grads, 2) != 0) {
|
|
fail();
|
|
}
|
|
|
|
printf("get updated parameters from pserver:\n");
|
|
// get parameters again by reusing the allocated parameter buffers.
|
|
if (paddle_get_params(c, params, 2) != 0) {
|
|
fail();
|
|
}
|
|
print_parameter(params[0]);
|
|
print_parameter(params[1]);
|
|
|
|
if (paddle_save_model(c, "/tmp/") != 0) {
|
|
fail();
|
|
}
|
|
|
|
return 0;
|
|
}
|