mirror/eigen
2
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2024-12-21 07:19:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed the type casting benchmark for float16

Browse Source
This commit is contained in:
Benoit Steiner 2016-04-07 22:45:54 -07:00
parent 7d5b17087f
commit 166b56bc61
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
bench/tensors/tensor_benchmarks.h
View File

@ -48,6 +48,10 @@ template <typename Device, typename T> class BenchmarkSuite {
Eigen::array<TensorIndex, 2> sizes; Eigen::array<TensorIndex, 2> sizes;
sizes[0] = m_; sizes[0] = m_;
sizes[1] = k_; sizes[1] = k_;
if (sizeof(T) < sizeof(int)) {
sizes[0] = m_ * sizeof(T) / sizeof(int);
sizes[1] = k_ * sizeof(T) / sizeof(int);
}
const TensorMap<Tensor<int, 2, 0, TensorIndex>, Eigen::Aligned> A((int*)a_, 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); TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(b_, sizes);