This repository was archived by the owner on Apr 28, 2023. It is now read-only.
File tree Expand file tree Collapse file tree 3 files changed +30
-31
lines changed Expand file tree Collapse file tree 3 files changed +30
-31
lines changed Original file line number Diff line number Diff line change @@ -69,8 +69,8 @@ WITH_CAFFE2=ON CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda CLANG_PREFIX=$(${CONDA_PREF
6969python setup.py install
7070./test_python/run_test.sh
7171
72- FILTER_OUT=" MLP_model kronecker " ./test.sh
72+ FILTER_OUT=" benchmark_MLP_model benchmark_kronecker " ./test.sh
7373# 2LUT can OOM on smaller Maxwells on our CI machines
74- ./build/tc/benchmarks/MLP_model --gtest_filter=-* 2LUT*
74+ ./build/tc/benchmarks/benchmark_MLP_model --gtest_filter=-* 2LUT*
7575# Kronecker xxxAsMatMul can OOM
76- ./build/tc/benchmarks/kronecker --gtest_filter=-* AsMatMul*
76+ ./build/tc/benchmarks/benchmark_kronecker --gtest_filter=-* AsMatMul*
Original file line number Diff line number Diff line change @@ -13,20 +13,19 @@ find_library(CUDA_CUDNN_LIBRARIES cudnn
1313 PATH_SUFFIXES lib lib64)
1414
1515################################################################################
16- # Examples
16+ # Benchmarks
1717################################################################################
18- set (EXAMPLES_FILES
18+ set (BENCHMARKS
1919 batchmatmul
2020 group_convolution
2121 kronecker
2222 tmm
2323 MLP_model
2424)
25- foreach (i ${EXAMPLES_FILES} )
26- add_executable (${i} ${i} .cc)
27- add_test (${i} ${i} )
25+ foreach (i ${BENCHMARKS} )
26+ add_executable (benchmark_${i} ${i} .cc)
2827 target_link_libraries (
29- ${i}
28+ benchmark_ ${i}
3029
3130 ${CUDA_CUBLAS_LIBRARIES}
3231 ${CUDA_CUDNN_LIBRARIES}
You can’t perform that action at this time.
0 commit comments