From edc482a20213ca806ae685f42467cd20e9852aa5 Mon Sep 17 00:00:00 2001 From: jiangcheng Date: Thu, 4 Aug 2022 10:40:37 +0800 Subject: [PATCH] add DCINN_GIT_TAG cmake flag to set cinn version (#44868) --- cmake/external/cinn.cmake | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/cmake/external/cinn.cmake b/cmake/external/cinn.cmake index 5dd84657c86..c09ee715c30 100644 --- a/cmake/external/cinn.cmake +++ b/cmake/external/cinn.cmake @@ -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} -- GitLab