提交 82740b1f 编写于 作者: T tianshuo78520a

notest;test=kunlun

上级 365d89db
......@@ -895,7 +895,11 @@ function card_test() {
case_count $1 $2
ut_startTime_s=`date +%s`
# get the CUDA device count
CUDA_DEVICE_COUNT=$(nvidia-smi -L | wc -l)
if [ "${WITH_XPU}" == "ON" ];then
CUDA_DEVICE_COUNT=$(xpu_smi -m 2 | wc -l)
else
CUDA_DEVICE_COUNT=$(nvidia-smi -L | wc -l)
fi
testcases=$1
if (( $# > 1 )); then
......@@ -1173,7 +1177,8 @@ function parallel_test_base_xpu() {
========================================
EOF
ut_startTime_s=`date +%s`
ctest -R *xpu
exclusive_tests=$(ctest -N |grep "_xpu") # cases list which would be run exclusively
card_test $exclusive_tests 1
ut_endTime_s=`date +%s`
echo "XPU testCase Time: $[ $ut_endTime_s - $ut_startTime_s ]s"
if [[ "$EXIT_CODE" != "0" ]]; then
......
......@@ -546,6 +546,10 @@ add_subdirectory(sequence)
add_subdirectory(dygraph_to_static)
add_subdirectory(rnn)
if (WITH_XPU)
add_subdirectory(xpu)
endif()
if (WITH_MKLDNN)
add_subdirectory(mkldnn)
endif()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册