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.
114 lines
2.9 KiB
114 lines
2.9 KiB
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
|
|
#include "libpaddle_pserver_cclient.h"
|
|
|
|
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);
|
|
for (int i = 0; i < param->content_len; ++i) {
|
|
printf("0x%x ", param->content[i]);
|
|
}
|
|
printf("\n\n");
|
|
}
|
|
}
|
|
|
|
int main() {
|
|
char addr[] = "localhost:3000";
|
|
client c = paddle_new_pserver_client(addr, 1);
|
|
retry:
|
|
if (paddle_begin_init_params(c)) {
|
|
paddle_parameter param;
|
|
char name_a[] = "param_a";
|
|
char name_b[] = "param_b";
|
|
unsigned char content1[] = {0x01, 0x02, 0x03};
|
|
param.element_type = PADDLE_ELEMENT_TYPE_FLOAT32;
|
|
param.name = name_a;
|
|
param.content = content1;
|
|
param.content_len = 3;
|
|
if (paddle_init_param(c, param, NULL, 0) != 0) {
|
|
goto retry;
|
|
}
|
|
unsigned char content2[] = {0x04, 0x05, 0x06};
|
|
param.element_type = PADDLE_ELEMENT_TYPE_INT32;
|
|
param.name = name_b;
|
|
param.content = content2;
|
|
param.content_len = 3;
|
|
if (paddle_init_param(c, param, NULL, 0) != 0) {
|
|
goto retry;
|
|
}
|
|
|
|
if (paddle_finish_init_params(c) != 0) {
|
|
goto retry;
|
|
}
|
|
} else {
|
|
fail();
|
|
}
|
|
|
|
unsigned char content1[] = {0x12, 0x23, 0x34};
|
|
unsigned char content2[] = {0x45, 0x56, 0x67};
|
|
|
|
paddle_gradient** new_params =
|
|
(paddle_gradient**)malloc(sizeof(paddle_gradient*) * 2);
|
|
new_params[0] = (paddle_gradient*)malloc(sizeof(paddle_gradient));
|
|
new_params[0]->name = "param_a";
|
|
new_params[0]->content = content1;
|
|
new_params[0]->content_len = 3;
|
|
new_params[0]->element_type = PADDLE_ELEMENT_TYPE_FLOAT32;
|
|
|
|
new_params[1] = (paddle_gradient*)malloc(sizeof(paddle_gradient));
|
|
new_params[1]->name = "param_b";
|
|
new_params[1]->content = content2;
|
|
new_params[1]->content_len = 3;
|
|
new_params[1]->element_type = PADDLE_ELEMENT_TYPE_INT32;
|
|
|
|
print_parameter(new_params[0]);
|
|
print_parameter(new_params[1]);
|
|
|
|
if (paddle_send_grads(c, new_params, 2) != 0) {
|
|
fail();
|
|
}
|
|
|
|
paddle_parameter* params[2] = {NULL, NULL};
|
|
char* names[] = {"param_a", "param_b"};
|
|
if (paddle_get_params(c, names, params, 2) != 0) {
|
|
fail();
|
|
}
|
|
|
|
print_parameter(params[0]);
|
|
print_parameter(params[1]);
|
|
|
|
/// change name of parameter.
|
|
char* names2[] = {"param_1", "param_2"};
|
|
if (paddle_get_params(c, names2, params, 2) == 0) {
|
|
fail();
|
|
}
|
|
|
|
// get parameters again by reusing the allocated parameter buffers.
|
|
if (paddle_get_params(c, names, params, 2) != 0) {
|
|
fail();
|
|
}
|
|
|
|
paddle_release_param(params[0]);
|
|
paddle_release_param(params[1]);
|
|
|
|
if (paddle_save_model(c, "/tmp/") != 0) {
|
|
fail();
|
|
}
|
|
|
|
printf("test success!\n");
|
|
return 0;
|
|
}
|