aboutsummaryrefslogtreecommitdiffhomepage
path: root/bench/tensors/README
diff options
context:
space:
mode:
authorGravatar Benoit Steiner <benoit.steiner.goog@gmail.com>2016-02-26 12:21:25 -0800
committerGravatar Benoit Steiner <benoit.steiner.goog@gmail.com>2016-02-26 12:21:25 -0800
commit002824e32def5c9a430acac4bd9fc05308c923bb (patch)
treebb669ed87de715400ff87a6623a7d188b68da4fa /bench/tensors/README
parent2cd32cad2782c467316297cec1a2b0ddff89c686 (diff)
Added benchmarks for fp16
Diffstat (limited to 'bench/tensors/README')
-rw-r--r--bench/tensors/README8
1 files changed, 6 insertions, 2 deletions
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
+