Skip to content

Commit 9e0a03e

Browse files
Add CUDA_ARCH_BIN (PaddlePaddle#38569)
1 parent ea01e79 commit 9e0a03e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

paddle/scripts/paddle_build.sh

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -236,6 +236,7 @@ function cmake_base() {
236236
-DON_INFER=${ON_INFER:-OFF}
237237
-DWITH_HETERPS=${WITH_HETERPS:-OFF}
238238
-DWITH_FLUID_ONLY=${WITH_FLUID_ONLY:-OFF}
239+
-DCUDA_ARCH_BIN="${DCUDA_ARCH_BIN}"
239240
========================================
240241
EOF
241242
# Disable UNITTEST_USE_VIRTUALENV in docker because
@@ -283,6 +284,7 @@ EOF
283284
-DON_INFER=${ON_INFER:-OFF} \
284285
-DWITH_HETERPS=${WITH_HETERPS:-OFF} \
285286
-DWITH_FLUID_ONLY=${WITH_FLUID_ONLY:-OFF} \
287+
-DCUDA_ARCH_BIN="${DCUDA_ARCH_BIN}" \
286288
-DWITH_UNITY_BUILD=${WITH_UNITY_BUILD:-OFF};build_error=$?
287289
if [ "$build_error" != 0 ];then
288290
exit 7;

0 commit comments

Comments
 (0)