Merge pull request #423 from nicolasvasilache/pr/benchmarks

Benchmarks

Author: ftynse

.jenkins/build.sh

@@ -69,5 +69,8 @@ WITH_CAFFE2=ON CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda CLANG_PREFIX=$(${CONDA_PREF
 python setup.py install
 ./test_python/run_test.sh
 
-FILTER_OUT=MLP_model ./test.sh
-./build/tc/benchmarks/MLP_model --gtest_filter=-*2LUT*
+FILTER_OUT="benchmark_MLP_model benchmark_kronecker" ./test.sh
+# 2LUT can OOM on smaller Maxwells on our CI machines
+./build/tc/benchmarks/benchmark_MLP_model --gtest_filter=-*2LUT*
+# Kronecker xxxAsMatMul can OOM
+./build/tc/benchmarks/benchmark_kronecker --gtest_filter=-*AsMatMul*

tc/benchmarks/CMakeLists.txt

@@ -13,19 +13,21 @@ find_library(CUDA_CUDNN_LIBRARIES cudnn
   PATH_SUFFIXES lib lib64)
 
 ################################################################################
-# Examples
+# Benchmarks
 ################################################################################
-set(EXAMPLES_FILES
+set(BENCHMARKS
   batchmatmul
   group_convolution
+  kronecker
+  moments
   tmm
   MLP_model
+  wavenet
 )
-foreach(i ${EXAMPLES_FILES})
-  add_executable(${i} ${i}.cc)
-  add_test(${i} ${i})
+foreach(i ${BENCHMARKS})
+  add_executable(benchmark_${i} ${i}.cc)
   target_link_libraries(
-     ${i}
+     benchmark_${i}
 
      ${CUDA_CUBLAS_LIBRARIES}
      ${CUDA_CUDNN_LIBRARIES}