diff --git a/cmake/external/cinn.cmake b/cmake/external/cinn.cmake index 004bf353d34e8ec0f19582663dfb18231152722e..2ec9a3faa07b7877efd242023c6a6e3c63933468 100644 --- a/cmake/external/cinn.cmake +++ b/cmake/external/cinn.cmake @@ -26,7 +26,7 @@ add_definitions(-w) ###################################### include(ExternalProject) set(CINN_PREFIX_DIR ${THIRD_PARTY_PATH}/CINN) -set(CINN_GIT_TAG eedb801ca39bfc6b9621bc76c24a0bf98cb8404b) +set(CINN_GIT_TAG release/v0.2) set(CINN_OPTIONAL_ARGS -DPY_VERSION=${PY_VERSION} -DWITH_CUDA=${WITH_GPU} -DWITH_CUDNN=${WITH_GPU}