eigen/bench/tensors
2016-02-26 12:24:58 -08:00
..
benchmark_main.cc Fixed the tensor benchmarks on apple devices 2016-01-28 21:08:07 -08:00
benchmark.h Updated the benchmarking code to print the number of flops processed instead of the number of bytes. 2016-01-28 16:51:40 -08:00
README Added benchmarks for fp16 2016-02-26 12:21:25 -08:00
tensor_benchmarks_cpu.cc Extended the tensor benchmark suite to support types other than floats 2016-02-23 05:28:02 +00:00
tensor_benchmarks_fp16_gpu.cu Added benchmarks for type casting of float16 2016-02-26 12:24:58 -08:00
tensor_benchmarks_gpu.cu Extended the tensor benchmark suite to support types other than floats 2016-02-23 05:28:02 +00:00
tensor_benchmarks.h Added benchmarks for type casting of float16 2016-02-26 12:24:58 -08:00

Each benchmark comes in 2 flavors: one that runs on CPU, and one that runs on GPU.

To compile the floating point CPU benchmarks, simply call:
g++ tensor_benchmarks_cpu.cc benchmark_main.cc -I ../../ -std=c++11 -O3 -DNDEBUG -pthread -mavx -o benchmarks_cpu

To compile the floating point GPU benchmarks, simply call:
nvcc tensor_benchmarks_gpu.cu benchmark_main.cc -I ../../ -std=c++11 -O2 -DNDEBUG -arch compute_35 -o benchmarks_gpu


To compile the half float GPU benchmarks, simply call the command line below. You'll need a recent GPU that supports compute capability 5.3 or higher to run them and nvcc 7.5 or higher to compile the code.
nvcc tensor_benchmarks_fp16_gpu.cu benchmark_main.cc -I ../../ -std=c++11 -O2 -DNDEBUG -arch compute_53 -o benchmarks_gpu