From 93485d86bcddc0665939ce2c43261dfaa1b8cc90 Mon Sep 17 00:00:00 2001
From: Benoit Steiner <benoit.steiner.goog@gmail.com>
Date: Fri, 26 Feb 2016 12:24:58 -0800
Subject: Added benchmarks for type casting of float16

---
 bench/tensors/tensor_benchmarks.h           | 6 +++---
 bench/tensors/tensor_benchmarks_fp16_gpu.cu | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

(limited to 'bench/tensors')

diff --git a/bench/tensors/tensor_benchmarks.h b/bench/tensors/tensor_benchmarks.h
index b208a401a..131d056b4 100644
--- a/bench/tensors/tensor_benchmarks.h
+++ b/bench/tensors/tensor_benchmarks.h
@@ -48,12 +48,12 @@ template <typename Device, typename T> class BenchmarkSuite {
     Eigen::array<TensorIndex, 2> sizes;
     sizes[0] = m_;
     sizes[1] = k_;
-    const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> A(a_, sizes);
-    TensorMap<Tensor<int, 2, 0, TensorIndex>, Eigen::Aligned> B((int*)b_, sizes);
+    const TensorMap<Tensor<int, 2, 0, TensorIndex>, Eigen::Aligned> A((int*)a_, sizes);
+    TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, sizes);
 
     StartBenchmarkTiming();
     for (int iter = 0; iter < num_iters; ++iter) {
-      B.device(device_) = A.template cast<int>();
+      B.device(device_) = A.template cast<T>();
     }
     // Record the number of values copied per second
     finalizeBenchmark(static_cast<int64_t>(m_) * k_ * num_iters);
diff --git a/bench/tensors/tensor_benchmarks_fp16_gpu.cu b/bench/tensors/tensor_benchmarks_fp16_gpu.cu
index d841bcdac..49f75472a 100644
--- a/bench/tensors/tensor_benchmarks_fp16_gpu.cu
+++ b/bench/tensors/tensor_benchmarks_fp16_gpu.cu
@@ -19,7 +19,7 @@
   BENCHMARK_RANGE(BM_##FUNC, 10, 5000);
 
 BM_FuncGPU(memcpy);
-//BM_FuncGPU(typeCasting);
+BM_FuncGPU(typeCasting);
 //BM_FuncGPU(random);
 BM_FuncGPU(slicing);
 BM_FuncGPU(rowChip);
-- 
cgit v1.2.3