diff options
| author |  Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-02-23 05:28:02 +0000 |
|---|---|---|
| committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-02-23 05:28:02 +0000 |
| commit | 8cb9bfab870c1f55ea9c69233a832e92c8de189d (patch) | |
| tree | a06ca43a0b4e4404b45a41f6d92cef2e03153f60 /bench/tensors/tensor_benchmarks_gpu.cu | |
| parent | f442a5a5b34ede4ab4e8fe36d1c8237315ad3f04 (diff) | |
Extended the tensor benchmark suite to support types other than floats
Diffstat (limited to 'bench/tensors/tensor_benchmarks_gpu.cu')
| -rw-r--r-- | bench/tensors/tensor_benchmarks_gpu.cu | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/bench/tensors/tensor_benchmarks_gpu.cu b/bench/tensors/tensor_benchmarks_gpu.cu index 611e8197b..a6f594382 100644 --- a/bench/tensors/tensor_benchmarks_gpu.cu +++ b/bench/tensors/tensor_benchmarks_gpu.cu @@ -12,7 +12,7 @@ StopBenchmarkTiming(); \ Eigen::CudaStreamDevice stream; \ Eigen::GpuDevice device(&stream); \ - BenchmarkSuite<Eigen::GpuDevice> suite(device, N); \ + BenchmarkSuite<Eigen::GpuDevice, float> suite(device, N); \ cudaDeviceSynchronize(); \ suite.FUNC(iters); \ } \ @@ -41,7 +41,7 @@ BM_FuncGPU(colReduction); StopBenchmarkTiming(); \ Eigen::CudaStreamDevice stream; \ Eigen::GpuDevice device(&stream); \ - BenchmarkSuite<Eigen::GpuDevice> suite(device, D1, D2, D3); \ + BenchmarkSuite<Eigen::GpuDevice, float> suite(device, D1, D2, D3); \ cudaDeviceSynchronize(); \ suite.FUNC(iters); \ } \ @@ -60,7 +60,7 @@ BM_FuncWithInputDimsGPU(contraction, N, N, 64); StopBenchmarkTiming(); \ Eigen::CudaStreamDevice stream; \ Eigen::GpuDevice device(&stream); \ - BenchmarkSuite<Eigen::GpuDevice> suite(device, N); \ + BenchmarkSuite<Eigen::GpuDevice, float> suite(device, N); \ cudaDeviceSynchronize(); \ suite.FUNC(iters, DIM1, DIM2); \ } \ |