diff --git a/cmake/external/cinn.cmake b/cmake/external/cinn.cmake index 5dd84657c86054e072346c6d060d2b7facd8adc1..c09ee715c304330f304b28706ebc7c1d50b7e3e0 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}