diff options
| author |  Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-02-24 17:07:25 -0800 |
|---|---|---|
| committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-02-24 17:07:25 -0800 |
| commit | c36c09169e1545e287293f3f145fa5a25b47b84a (patch) | |
| tree | ea3166ffbc92b1ca646dad942c80390899a04296 /unsupported/Eigen/CXX11/src/Tensor/TensorReduction.h | |
| parent | 7a01cb8e4b647e8cf4223f30266c8bab1d3fb66c (diff) | |
Fixed a typo in the reduction code that could prevent large full reductionsx from running properly on old cuda devices.
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorReduction.h')
| -rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorReduction.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorReduction.h b/unsupported/Eigen/CXX11/src/Tensor/TensorReduction.h index 22aea5ea4..f7c1a5cf4 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorReduction.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorReduction.h @@ -515,7 +515,7 @@ struct TensorEvaluator<const TensorReductionOp<Op, Dims, ArgType>, Device> // Use the FullReducer if possible. if (RunningFullReduction && internal::FullReducer<Self, Op, Device>::HasOptimizedImplementation && ((RunningOnGPU && (m_device.majorDeviceVersion() >= 3)) || - (internal::array_prod(m_impl.dimensions()) > 1024 * 1024))) { + (!RunningOnGPU && (internal::array_prod(m_impl.dimensions()) > 1024 * 1024)))) { bool need_assign = false; if (!data) { |