eigen/bench/tensors/README

13 lines
770 B
Plaintext
Raw Normal View History

Each benchmark comes in 2 flavors: one that runs on CPU, and one that runs on GPU.
2016-02-27 04:21:25 +08:00
To compile the floating point CPU benchmarks, simply call:
2016-01-29 09:13:04 +08:00
g++ tensor_benchmarks_cpu.cc benchmark_main.cc -I ../../ -std=c++11 -O3 -DNDEBUG -pthread -mavx -o benchmarks_cpu
2016-02-27 04:21:25 +08:00
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
2016-02-27 04:21:25 +08:00
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.
2016-02-28 04:22:04 +08:00
nvcc tensor_benchmarks_fp16_gpu.cu benchmark_main.cc -I ../../ -std=c++11 -O2 -DNDEBUG -arch compute_53 -o benchmarks_fp16_gpu
2016-02-27 04:21:25 +08:00