From 002824e32def5c9a430acac4bd9fc05308c923bb Mon Sep 17 00:00:00 2001 From: Benoit Steiner Date: Fri, 26 Feb 2016 12:21:25 -0800 Subject: Added benchmarks for fp16 --- bench/tensors/README | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'bench/tensors/README') diff --git a/bench/tensors/README b/bench/tensors/README index 6b51fe878..1de0a5786 100644 --- a/bench/tensors/README +++ b/bench/tensors/README @@ -1,8 +1,12 @@ Each benchmark comes in 2 flavors: one that runs on CPU, and one that runs on GPU. -To compile the CPU benchmarks, simply call: +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 GPU benchmarks, simply call: +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 + -- cgit v1.2.3