diff options
| author |  Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-03-24 18:00:33 -0700 |
|---|---|---|
| committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-03-24 18:00:33 -0700 |
| commit | 0968e925a040d4988f02e8476b5cea8518e5f966 (patch) | |
| tree | 09f5d718e6fe579f43a023d545fac38df692dc6d /bench/tensors | |
| parent | 044efea965b484fcf13551c8edabdb62c4b4b462 (diff) | |
Updated the benchmarking code to use Eigen::half instead of half
Diffstat (limited to 'bench/tensors')
| -rw-r--r-- | bench/tensors/tensor_benchmarks_fp16_gpu.cu | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/bench/tensors/tensor_benchmarks_fp16_gpu.cu b/bench/tensors/tensor_benchmarks_fp16_gpu.cu index 49f75472a..35c6f7489 100644 --- a/bench/tensors/tensor_benchmarks_fp16_gpu.cu +++ b/bench/tensors/tensor_benchmarks_fp16_gpu.cu @@ -12,7 +12,7 @@ StopBenchmarkTiming(); \ Eigen::CudaStreamDevice stream; \ Eigen::GpuDevice device(&stream); \ - BenchmarkSuite<Eigen::GpuDevice, half> suite(device, N); \ + BenchmarkSuite<Eigen::GpuDevice, Eigen::half> 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, half> suite(device, D1, D2, D3); \ + BenchmarkSuite<Eigen::GpuDevice, Eigen::half> 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, half> suite(device, N); \ + BenchmarkSuite<Eigen::GpuDevice, Eigen::half> suite(device, N); \ cudaDeviceSynchronize(); \ suite.FUNC(iters, DIM1, DIM2); \ } \ @@ -73,4 +73,4 @@ BM_FuncWithKernelDimsGPU(convolution, 7, 4); BM_FuncWithKernelDimsGPU(convolution, 4, 7); BM_FuncWithKernelDimsGPU(convolution, 7, 64); BM_FuncWithKernelDimsGPU(convolution, 64, 7); -*/
\ No newline at end of file +*/ |