Skip to content

Commit 5d3e816

Browse files
bug fix in dense
1 parent 67bd4cd commit 5d3e816

File tree

2 files changed

+14
-9
lines changed

2 files changed

+14
-9
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_property(TARGET ${PROJECT_NAME} PROPERTY C_STANDARD 99)
5+
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")
66
target_link_libraries(${PROJECT_NAME} -lpaddle_capi_shared)

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

Lines changed: 13 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,15 @@
11
#include <paddle/capi.h>
22
#include <time.h>
3+
#include <iostream>
4+
#include <vector>
35
#include "../common/common.h"
46

57
#define CONFIG_BIN "./trainer_config.bin"
68

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

1214
// Reading config binary file. It is generated by `convert_protobin.sh`
1315
long size;
@@ -53,17 +55,20 @@ int main() {
5355

5456
CHECK(paddle_arguments_get_value(out_args, 0, prob));
5557

56-
std::std::vector<paddle_real> result;
57-
int height;
58-
int width;
58+
std::vector<paddle_real> result;
59+
uint64_t height;
60+
uint64_t width;
5961

60-
CHECK(paddle_matrix_get_shape(prob, &height, &width);
62+
CHECK(paddle_matrix_get_shape(prob, &height, &width));
6163
result.resize(height * width);
6264
CHECK(paddle_matrix_get_value(prob, result.data()));
6365

64-
printf("Prob: ");
66+
printf("Prob: \n");
6567
for (int i = 0; i < height * width; ++i) {
66-
printf("%.2f ", result[i]);
68+
printf("%.4f ", result[i]);
69+
if ((i + 1) % width == 0){
70+
printf("\n");
71+
}
6772
}
6873
printf("\n");
6974

0 commit comments

Comments
 (0)