未验证 提交 9e0a03ee 编写于 作者: T tianshuo78520a 提交者: GitHub

Add CUDA_ARCH_BIN (#38569)

上级 ea01e790
......@@ -236,6 +236,7 @@ function cmake_base() {
-DON_INFER=${ON_INFER:-OFF}
-DWITH_HETERPS=${WITH_HETERPS:-OFF}
-DWITH_FLUID_ONLY=${WITH_FLUID_ONLY:-OFF}
-DCUDA_ARCH_BIN="${DCUDA_ARCH_BIN}"
========================================
EOF
# Disable UNITTEST_USE_VIRTUALENV in docker because
......@@ -283,6 +284,7 @@ EOF
-DON_INFER=${ON_INFER:-OFF} \
-DWITH_HETERPS=${WITH_HETERPS:-OFF} \
-DWITH_FLUID_ONLY=${WITH_FLUID_ONLY:-OFF} \
-DCUDA_ARCH_BIN="${DCUDA_ARCH_BIN}" \
-DWITH_UNITY_BUILD=${WITH_UNITY_BUILD:-OFF};build_error=$?
if [ "$build_error" != 0 ];then
exit 7;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册