Skip to content

Commit 82aaceb

Browse files
transform to c style
1 parent d6bd5b1 commit 82aaceb

File tree

2 files changed

+12
-17
lines changed

2 files changed

+12
-17
lines changed

paddle/capi/examples/model_inference/dense/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,5 +2,5 @@ project(dense)
22
cmake_minimum_required(VERSION 2.8)
33
aux_source_directory(. SRC_LIST)
44
add_executable(${PROJECT_NAME} ${SRC_LIST})
5-
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
5+
set_property(TARGET ${PROJECT_NAME} PROPERTY C_STANDARD 99)
66
target_link_libraries(${PROJECT_NAME} -lpaddle_capi_shared)

paddle/capi/examples/model_inference/dense/main.cpp renamed to paddle/capi/examples/model_inference/dense/main.c

Lines changed: 11 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,14 @@
11
#include <paddle/capi.h>
22
#include <time.h>
3-
#include <iostream>
4-
#include <vector>
53

64
#include "../common/common.h"
75

86
#define CONFIG_BIN "./trainer_config.bin"
97

108
int main() {
119
// Initalize Paddle
12-
std::string comand[] = {"--use_gpu=False"};
13-
CHECK(paddle_init(1, (char**)comand));
10+
char* argv[] = {"--use_gpu=False"};
11+
CHECK(paddle_init(1, (char**)argv));
1412

1513
// Reading config binary file. It is generated by `convert_protobin.sh`
1614
long size;
@@ -30,20 +28,19 @@ int main() {
3028
CHECK(paddle_arguments_resize(in_args, 1));
3129

3230
// Create input matrix.
33-
paddle_matrix mat = paddle_matrix_create(/* sample_num */ 10,
31+
paddle_matrix mat = paddle_matrix_create(/* sample_num */ 1,
3432
/* size */ 784,
3533
/* useGPU */ false);
3634
srand(time(0));
3735

38-
std::vector<paddle_real> input;
39-
input.resize(784 * 10);
36+
paddle_real* array;
4037

41-
for (int i = 0; i < input.size(); ++i) {
42-
input[i] = rand() / ((float)RAND_MAX);
43-
}
38+
// Get First row.
39+
CHECK(paddle_matrix_get_row(mat, 0, &array));
4440

45-
// Set value for the input matrix
46-
CHECK(paddle_matrix_set_value(mat, input.data()));
41+
for (int i = 0; i < 784; ++i) {
42+
array[i] = rand() / ((float)RAND_MAX);
43+
}
4744

4845
CHECK(paddle_arguments_set_value(in_args, 0, mat));
4946

@@ -56,17 +53,15 @@ int main() {
5653

5754
CHECK(paddle_arguments_get_value(out_args, 0, prob));
5855

59-
std::vector<paddle_real> result;
6056
uint64_t height;
6157
uint64_t width;
6258

6359
CHECK(paddle_matrix_get_shape(prob, &height, &width));
64-
result.resize(height * width);
65-
CHECK(paddle_matrix_get_value(prob, result.data()));
60+
CHECK(paddle_matrix_get_row(prob, 0, &array));
6661

6762
printf("Prob: \n");
6863
for (int i = 0; i < height * width; ++i) {
69-
printf("%.4f ", result[i]);
64+
printf("%.4f ", array[i]);
7065
if ((i + 1) % width == 0) {
7166
printf("\n");
7267
}

0 commit comments

Comments
 (0)