未验证 提交 40cd9552 编写于 作者: Y YUNSHEN XIE 提交者: GitHub

add cinn_git_tag param (#50539)

* add cinn_git_tag param

* fix format error
上级 4230bd87
...@@ -201,6 +201,7 @@ function cmake_base() { ...@@ -201,6 +201,7 @@ function cmake_base() {
-DWITH_TENSORRT=${WITH_TENSORRT:-ON} -DWITH_TENSORRT=${WITH_TENSORRT:-ON}
-DWITH_ROCM=${WITH_ROCM:-OFF} -DWITH_ROCM=${WITH_ROCM:-OFF}
-DWITH_CINN=${WITH_CINN:-OFF} -DWITH_CINN=${WITH_CINN:-OFF}
-DCINN_GIT_TAG=${CINN_GIT_TAG:-develop}
-DWITH_DISTRIBUTE=${distibuted_flag} -DWITH_DISTRIBUTE=${distibuted_flag}
-DWITH_MKL=${WITH_MKL:-ON} -DWITH_MKL=${WITH_MKL:-ON}
-DWITH_AVX=${WITH_AVX:-OFF} -DWITH_AVX=${WITH_AVX:-OFF}
...@@ -257,6 +258,7 @@ EOF ...@@ -257,6 +258,7 @@ EOF
-DWITH_TENSORRT=${WITH_TENSORRT:-ON} \ -DWITH_TENSORRT=${WITH_TENSORRT:-ON} \
-DWITH_ROCM=${WITH_ROCM:-OFF} \ -DWITH_ROCM=${WITH_ROCM:-OFF} \
-DWITH_CINN=${WITH_CINN:-OFF} \ -DWITH_CINN=${WITH_CINN:-OFF} \
-DCINN_GIT_TAG=${CINN_GIT_TAG:-develop} \
-DWITH_DISTRIBUTE=${distibuted_flag} \ -DWITH_DISTRIBUTE=${distibuted_flag} \
-DWITH_MKL=${WITH_MKL:-ON} \ -DWITH_MKL=${WITH_MKL:-ON} \
-DWITH_AVX=${WITH_AVX:-OFF} \ -DWITH_AVX=${WITH_AVX:-OFF} \
...@@ -3628,6 +3630,7 @@ function run_setup(){ ...@@ -3628,6 +3630,7 @@ function run_setup(){
export WITH_TENSORRT=${WITH_TENSORRT:-ON} export WITH_TENSORRT=${WITH_TENSORRT:-ON}
export WITH_ROCM=${WITH_ROCM:-OFF} export WITH_ROCM=${WITH_ROCM:-OFF}
export WITH_CINN=${WITH_CINN:-OFF} export WITH_CINN=${WITH_CINN:-OFF}
export CINN_GIT_TAG=${CINN_GIT_TAG:-develop}
export WITH_DISTRIBUTE=${distibuted_flag} export WITH_DISTRIBUTE=${distibuted_flag}
export WITH_MKL=${WITH_MKL:-ON} export WITH_MKL=${WITH_MKL:-ON}
export WITH_AVX=${WITH_AVX:-OFF} export WITH_AVX=${WITH_AVX:-OFF}
...@@ -3858,6 +3861,7 @@ function run_setup_mac(){ ...@@ -3858,6 +3861,7 @@ function run_setup_mac(){
export WITH_TENSORRT=${WITH_TENSORRT:-ON} export WITH_TENSORRT=${WITH_TENSORRT:-ON}
export WITH_ROCM=${WITH_ROCM:-OFF} export WITH_ROCM=${WITH_ROCM:-OFF}
export WITH_CINN=${WITH_CINN:-OFF} export WITH_CINN=${WITH_CINN:-OFF}
export CINN_GIT_TAG=${CINN_GIT_TAG:-develop}
export WITH_DISTRIBUTE=${distibuted_flag} export WITH_DISTRIBUTE=${distibuted_flag}
export WITH_MKL=${WITH_MKL:-ON} export WITH_MKL=${WITH_MKL:-ON}
export WITH_AVX=${WITH_AVX:-OFF} export WITH_AVX=${WITH_AVX:-OFF}
......
...@@ -667,6 +667,7 @@ def cmake_run(build_path): ...@@ -667,6 +667,7 @@ def cmake_run(build_path):
"MSVC_STATIC_CRT", "MSVC_STATIC_CRT",
"NEW_RELEASE_ALL", "NEW_RELEASE_ALL",
"GENERATOR", "GENERATOR",
"CINN_GIT_TAG",
) )
} }
) )
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册