aboutsummaryrefslogtreecommitdiffhomepage
path: root/bench/tensors/tensor_benchmarks.h
diff options
context:
space:
mode:
authorGravatar Yangqing Jia <me@daggerfs.com>2016-01-28 11:11:45 -0800
committerGravatar Yangqing Jia <me@daggerfs.com>2016-01-28 11:11:45 -0800
commit270c4e1ecd8fd10c42760dd67adbbab0b1387da2 (patch)
tree91ff1a778419cee6cc99567d01680670ec0bc26b /bench/tensors/tensor_benchmarks.h
parentc4e47630b16a716d01dc20b36afa8882b03681a1 (diff)
bugfix
Diffstat (limited to 'bench/tensors/tensor_benchmarks.h')
-rw-r--r--bench/tensors/tensor_benchmarks.h13
1 files changed, 3 insertions, 10 deletions
diff --git a/bench/tensors/tensor_benchmarks.h b/bench/tensors/tensor_benchmarks.h
index a1696afda..071326aa7 100644
--- a/bench/tensors/tensor_benchmarks.h
+++ b/bench/tensors/tensor_benchmarks.h
@@ -10,13 +10,6 @@ typedef int TensorIndex;
#define BENCHMARK_RANGE(bench, lo, hi) \
BENCHMARK(bench)->Range(lo, hi)
-template <typename... Args>
-std::string StrCat(const Args... args) {
- std::stringstream ss;
- StrCatRecursive(ss, args...);
- return ss.str();
-}
-
using Eigen::Tensor;
using Eigen::TensorMap;
@@ -305,9 +298,9 @@ template <typename Device> class BenchmarkSuite {
}
- size_t m_;
- size_t k_;
- size_t n_;
+ TensorIndex m_;
+ TensorIndex k_;
+ TensorIndex n_;
float* a_;
float* b_;
float* c_;