diff --git a/paddle/capi/examples/model_inference/dense/CMakeLists.txt b/paddle/capi/examples/model_inference/dense/CMakeLists.txt index 008a488fd9e6fdca2c4cb92bf1b8c41fce1835a9..31759310ce1c21071c53f4c4685c68495ed1cc93 100644 --- a/paddle/capi/examples/model_inference/dense/CMakeLists.txt +++ b/paddle/capi/examples/model_inference/dense/CMakeLists.txt @@ -2,5 +2,5 @@ project(dense) cmake_minimum_required(VERSION 2.8) aux_source_directory(. SRC_LIST) add_executable(${PROJECT_NAME} ${SRC_LIST}) -set_property(TARGET ${PROJECT_NAME} PROPERTY C_STANDARD 99) +set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11") target_link_libraries(${PROJECT_NAME} -lpaddle_capi_shared) diff --git a/paddle/capi/examples/model_inference/dense/main.c b/paddle/capi/examples/model_inference/dense/main.cpp similarity index 85% rename from paddle/capi/examples/model_inference/dense/main.c rename to paddle/capi/examples/model_inference/dense/main.cpp index 876af2aa7615c098d225b56ce2ea0b1529a6e3c6..e761dfe2b7709ca01d7c14dee85233ad08e451e2 100644 --- a/paddle/capi/examples/model_inference/dense/main.c +++ b/paddle/capi/examples/model_inference/dense/main.cpp @@ -1,13 +1,15 @@ #include #include +#include +#include #include "../common/common.h" #define CONFIG_BIN "./trainer_config.bin" int main() { // Initalize Paddle - char* argv[] = {"--use_gpu=False"}; - CHECK(paddle_init(1, (char**)argv)); + std::string comand[] = {"--use_gpu=False"}; + CHECK(paddle_init(1, (char**)comand)); // Reading config binary file. It is generated by `convert_protobin.sh` long size; @@ -53,17 +55,20 @@ int main() { CHECK(paddle_arguments_get_value(out_args, 0, prob)); - std::std::vector result; - int height; - int width; + std::vector result; + uint64_t height; + uint64_t width; - CHECK(paddle_matrix_get_shape(prob, &height, &width); + CHECK(paddle_matrix_get_shape(prob, &height, &width)); result.resize(height * width); CHECK(paddle_matrix_get_value(prob, result.data())); - printf("Prob: "); + printf("Prob: \n"); for (int i = 0; i < height * width; ++i) { - printf("%.2f ", result[i]); + printf("%.4f ", result[i]); + if ((i + 1) % width == 0){ + printf("\n"); + } } printf("\n");