未验证 提交 edc482a2 编写于 作者: J jiangcheng 提交者: GitHub

add DCINN_GIT_TAG cmake flag to set cinn version (#44868)

上级 7335b679
......@@ -16,6 +16,12 @@ if(NOT WITH_CINN)
return()
endif()
if(NOT CINN_GIT_TAG)
set(CINN_GIT_TAG release/v0.2)
endif()
message(STATUS "CINN version: " ${CINN_GIT_TAG})
# TODO(zhhsplendid): CINN has lots of warnings during early development.
# They will be treated as errors under paddle. We set no-error now and we will
# clean the code in the future.
......@@ -26,7 +32,6 @@ add_definitions(-w)
######################################
include(ExternalProject)
set(CINN_PREFIX_DIR ${THIRD_PARTY_PATH}/CINN)
set(CINN_GIT_TAG release/v0.2)
set(CINN_OPTIONAL_ARGS
-DPY_VERSION=${PY_VERSION}
-DWITH_CUDA=${WITH_GPU}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册