diff options
author | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-04-12 14:11:52 -0700 |
---|---|---|
committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-04-12 14:11:52 -0700 |
commit | 53121c0119d046228c2911c39cf33157479be88d (patch) | |
tree | db6edb7a9924e2377957b126e8851a513331c22b /bench/tensors | |
parent | e3a184785c3ea106b8f74471a325ea0945127639 (diff) |
Turned on the contraction benchmarks for fp16
Diffstat (limited to 'bench/tensors')
-rw-r--r-- | bench/tensors/tensor_benchmarks_fp16_gpu.cu | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/bench/tensors/tensor_benchmarks_fp16_gpu.cu b/bench/tensors/tensor_benchmarks_fp16_gpu.cu index d34bd73ca..d6719b7a2 100644 --- a/bench/tensors/tensor_benchmarks_fp16_gpu.cu +++ b/bench/tensors/tensor_benchmarks_fp16_gpu.cu @@ -48,11 +48,11 @@ BM_FuncGPU(colReduction); BENCHMARK_RANGE(BM_##FUNC##_##D1##x##D2##x##D3, 10, 5000); -/*BM_FuncWithInputDimsGPU(contraction, N, N, N); +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 #define BM_FuncWithKernelDimsGPU(FUNC, DIM1, DIM2) \ |