提交 e6b2f877 编写于 作者: Z zhangjun

update python CMakeLists.txt

上级 1adab94b
...@@ -62,6 +62,8 @@ if (CLIENT) ...@@ -62,6 +62,8 @@ if (CLIENT)
endif() endif()
if (SERVER) if (SERVER)
# todo, generate suffix for cpu、gpu、arm
if(WITH_TRT)
if(CUDA_VERSION EQUAL 10.1) if(CUDA_VERSION EQUAL 10.1)
set(VERSION_SUFFIX 101) set(VERSION_SUFFIX 101)
elseif(CUDA_VERSION EQUAL 10.2) elseif(CUDA_VERSION EQUAL 10.2)
...@@ -69,6 +71,11 @@ if (SERVER) ...@@ -69,6 +71,11 @@ if (SERVER)
elseif(CUDA_VERSION EQUAL 11.0) elseif(CUDA_VERSION EQUAL 11.0)
set(VERSION_SUFFIX 11) set(VERSION_SUFFIX 11)
endif() endif()
endif()
if(WITH_LITE)
set(VERSION_SUFFIX xpu)
endif()
add_custom_command( add_custom_command(
OUTPUT ${PADDLE_SERVING_BINARY_DIR}/.timestamp OUTPUT ${PADDLE_SERVING_BINARY_DIR}/.timestamp
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册