Skip to content

Commit b5f7886

Browse files
authored
[feature][win] enable paddle_inference in windows (#103)
* [feature][win] enable paddle_inference in windows * [feature][win] enable paddle_inference in windows * [feature][win] enable paddle_inference in windows
1 parent affbdf0 commit b5f7886

1 file changed

Lines changed: 7 additions & 4 deletions

File tree

external/paddle_inference.cmake

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,8 @@ if(WIN32)
3232
set(PADDLEINFERENCE_COMPILE_LIB
3333
"${PADDLEINFERENCE_INSTALL_DIR}/paddle/lib/paddle_inference.lib"
3434
CACHE FILEPATH "paddle_inference compile library." FORCE)
35-
set(DNNL_LIB "")
36-
set(OMP_LIB "")
35+
set(DNNL_LIB "${PADDLEINFERENCE_INSTALL_DIR}/third_party/install/mkldnn/lib/mkldnn.lib")
36+
set(OMP_LIB "${PADDLEINFERENCE_INSTALL_DIR}/third_party/install/mklml/lib/libiomp5md.lib")
3737
elseif(APPLE)
3838
set(PADDLEINFERENCE_COMPILE_LIB
3939
"${PADDLEINFERENCE_INSTALL_DIR}/paddle/lib/libpaddle_inference.dylib"
@@ -51,8 +51,11 @@ endif(WIN32)
5151
set(PADDLEINFERENCE_URL_BASE "https://bj.bcebos.com/paddle2onnx/libs/")
5252
set(PADDLEINFERENCE_VERSION "2.3.2")
5353
if(WIN32)
54-
message(FATAL_ERROR "Paddle Backend doesn't support Windows now.")
55-
set(PADDLEINFERENCE_FILE "paddle_inference-win-x64-${PADDLEINFERENCE_VERSION}.zip")
54+
if (WITH_GPU)
55+
set(PADDLEINFERENCE_FILE "paddle_inference-win-x64-gpu-${PADDLEINFERENCE_VERSION}.zip")
56+
else()
57+
set(PADDLEINFERENCE_FILE "paddle_inference-win-x64-${PADDLEINFERENCE_VERSION}.zip")
58+
endif()
5659
elseif(APPLE)
5760
message(FATAL_ERROR "Paddle Backend doesn't support Mac OSX now.")
5861
if(CMAKE_HOST_SYSTEM_PROCESSOR MATCHES "arm64")

0 commit comments

Comments
 (0)