diff options
| author |  Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-01-28 16:20:36 -0800 |
|---|---|---|
| committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-01-28 16:20:36 -0800 |
| commit | c8d5f21941a41556f94e937ea5a91badb7fb9353 (patch) | |
| tree | 163d3b29f6ce464803dfa5831aa3e31f91c8c7c8 /bench/tensors/tensor_benchmarks_gpu.cu | |
| parent | 7b3044d086f413fdaf65acd30fc3bc469d43ccc6 (diff) | |
Added extra tensor benchmarks
Diffstat (limited to 'bench/tensors/tensor_benchmarks_gpu.cu')
| -rw-r--r-- | bench/tensors/tensor_benchmarks_gpu.cu | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/bench/tensors/tensor_benchmarks_gpu.cu b/bench/tensors/tensor_benchmarks_gpu.cu index fbb486efd..fe807d2ab 100644 --- a/bench/tensors/tensor_benchmarks_gpu.cu +++ b/bench/tensors/tensor_benchmarks_gpu.cu @@ -19,6 +19,7 @@ BENCHMARK_RANGE(BM_##FUNC, 10, 5000); BM_FuncGPU(memcpy); +BM_FuncGPU(typeCasting); BM_FuncGPU(random); BM_FuncGPU(slicing); BM_FuncGPU(shuffling); @@ -26,7 +27,10 @@ BM_FuncGPU(padding); BM_FuncGPU(striding); BM_FuncGPU(broadcasting); BM_FuncGPU(coeffWiseOp); -BM_FuncGPU(reduction); +BM_FuncGPU(algebraicFunc); +BM_FuncGPU(transcendentalFunc); +BM_FuncGPU(rowReduction); +BM_FuncGPU(colReduction); // Contractions @@ -45,6 +49,7 @@ BM_FuncGPU(reduction); BM_FuncWithInputDimsGPU(contraction, N, N, N); BM_FuncWithInputDimsGPU(contraction, 64, N, N); BM_FuncWithInputDimsGPU(contraction, N, 64, N); +BM_FuncWithInputDimsGPU(contraction, N, N, 64); // Convolutions |