Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Commit de173b0

Browse files
committed
add the missing build_ubuntu_gpu_cuda101_cudnn7_mkldnn_cpp_test in runtime_functions.sh
1 parent d271348 commit de173b0

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

ci/docker/runtime_functions.sh

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -823,6 +823,26 @@ build_ubuntu_gpu_cuda101_cudnn7() {
823823
make cython PYTHON=python3
824824
}
825825

826+
build_ubuntu_gpu_cuda101_cudnn7_mkldnn_cpp_test() {
827+
set -ex
828+
export CC=gcc-7
829+
export CXX=g++-7
830+
build_ccache_wrappers
831+
make \
832+
USE_BLAS=openblas \
833+
USE_MKLDNN=1 \
834+
USE_CUDA=1 \
835+
USE_CUDA_PATH=/usr/local/cuda \
836+
USE_CUDNN=1 \
837+
USE_CPP_PACKAGE=1 \
838+
USE_DIST_KVSTORE=1 \
839+
CUDA_ARCH="$CI_CUDA_COMPUTE_CAPABILITIES" \
840+
USE_SIGNAL_HANDLER=1 \
841+
-j$(nproc)
842+
make test USE_CPP_PACKAGE=1 -j$(nproc)
843+
make cython PYTHON=python3
844+
}
845+
826846
build_ubuntu_amalgamation() {
827847
set -ex
828848
# Amalgamation can not be run with -j nproc

0 commit comments

Comments
 (0)