mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-03-07 18:27:40 +08:00
Made the tensor benchmarks compile on MacOS
This commit is contained in:
parent
2062ee2d26
commit
7168afde5e
@ -333,7 +333,7 @@ template <typename Device, typename T> class BenchmarkSuite {
|
|||||||
|
|
||||||
#ifndef EIGEN_HAS_INDEX_LIST
|
#ifndef EIGEN_HAS_INDEX_LIST
|
||||||
Eigen::array<TensorIndex, 1> sum_along_dim;
|
Eigen::array<TensorIndex, 1> sum_along_dim;
|
||||||
sum_along_dim = 1;
|
sum_along_dim[0] = 1;
|
||||||
#else
|
#else
|
||||||
// Take advantage of cxx11 to give the compiler information it can use to
|
// Take advantage of cxx11 to give the compiler information it can use to
|
||||||
// optimize the code.
|
// optimize the code.
|
||||||
@ -356,7 +356,7 @@ template <typename Device, typename T> class BenchmarkSuite {
|
|||||||
input_size[1] = n_;
|
input_size[1] = n_;
|
||||||
const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(
|
const TensorMap<Tensor<T, 2, 0, TensorIndex>, Eigen::Aligned> B(
|
||||||
b_, input_size);
|
b_, input_size);
|
||||||
const Eigen::array<TensorIndex, 0> output_size;
|
Eigen::array<TensorIndex, 0> output_size;
|
||||||
TensorMap<Tensor<float, 0, 0, TensorIndex>, Eigen::Aligned> C(
|
TensorMap<Tensor<float, 0, 0, TensorIndex>, Eigen::Aligned> C(
|
||||||
c_, output_size);
|
c_, output_size);
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user