From a6d08be9b2e1b9e11a488419b7dd0affcc321a32 Mon Sep 17 00:00:00 2001
From: Benoit Steiner <benoit.steiner.goog@gmail.com>
Date: Thu, 7 Apr 2016 17:13:44 -0700
Subject: Fixed the benchmarking of fp16 coefficient wise operations

---
 bench/tensors/tensor_benchmarks.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'bench')

diff --git a/bench/tensors/tensor_benchmarks.h b/bench/tensors/tensor_benchmarks.h
index a4f97728d..16b388abf 100644
--- a/bench/tensors/tensor_benchmarks.h
+++ b/bench/tensors/tensor_benchmarks.h
@@ -248,7 +248,7 @@ template <typename Device, typename T> class BenchmarkSuite {
 
     StartBenchmarkTiming();
     for (int iter = 0; iter < num_iters; ++iter) {
-      C.device(device_) = A * A.constant(3.14) + B * B.constant(2.7);
+      C.device(device_) = A * A.constant(static_cast<T>(3.14)) + B * B.constant(static_cast<T>(2.7));
     }
     // Record the number of FLOP executed per second (2 multiplications and
     // 1 addition per value)
-- 
cgit v1.2.3