diff options
author | Rasmus Munk Larsen <rmlarsen@google.com> | 2019-10-22 12:42:44 -0700 |
---|---|---|
committer | Rasmus Munk Larsen <rmlarsen@google.com> | 2019-10-22 12:42:44 -0700 |
commit | 97c0c5d485ddec0369326825a41db48d8505cf4c (patch) | |
tree | 9072616f37eacc24f407061ac74954d67da8c5ee /unsupported/Eigen/CXX11/src/Tensor/TensorBase.h | |
parent | 668ab3fc474e54c7919eda4fbaf11f3a99246494 (diff) |
Add block evaluation V2 to TensorAsyncExecutor.
Add async evaluation to a number of ops.
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorBase.h')
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorBase.h | 7 |
1 files changed, 1 insertions, 6 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h b/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h index a951bea6d..7aa98fac6 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorBase.h @@ -1129,16 +1129,11 @@ class TensorBase : public TensorBase<Derived, ReadOnlyAccessors> { return TensorDevice<Derived, DeviceType>(dev, derived()); } -#ifdef EIGEN_USE_THREADS // Select the async device on which to evaluate the expression. template <typename DeviceType, typename DoneCallback> - typename internal::enable_if< - internal::is_same<DeviceType, ThreadPoolDevice>::value, - TensorAsyncDevice<Derived, DeviceType, DoneCallback>>::type - device(const DeviceType& dev, DoneCallback done) { + TensorAsyncDevice<Derived, DeviceType, DoneCallback> device(const DeviceType& dev, DoneCallback done) { return TensorAsyncDevice<Derived, DeviceType, DoneCallback>(dev, derived(), std::move(done)); } -#endif // EIGEN_USE_THREADS protected: EIGEN_DEVICE_FUNC |