diff options
| author |  Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-11-20 10:12:54 -0800 |
|---|---|---|
| committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-11-20 10:12:54 -0800 |
| commit | 6d781e3e52b328710c55d854a97b2e69a3d433cc (patch) | |
| tree | 9045672a31964bde027421b4c52d414d29fce3a7 /unsupported/Eigen | |
| parent | 9265ca707ecde69c654029ff7209eff5cb50ffd4 (diff) | |
| parent | 79a07b891bed35eba8ba4267722ecf99bf16a9d8 (diff) | |
Merged eigen/eigen into default
Diffstat (limited to 'unsupported/Eigen')
| -rw-r--r-- | unsupported/Eigen/CXX11/Tensor | 1 | ||||
| -rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorDeviceSycl.h | 4 |
2 files changed, 5 insertions, 0 deletions
diff --git a/unsupported/Eigen/CXX11/Tensor b/unsupported/Eigen/CXX11/Tensor index e41b67c56..8b36093f0 100644 --- a/unsupported/Eigen/CXX11/Tensor +++ b/unsupported/Eigen/CXX11/Tensor @@ -20,6 +20,7 @@ #undef isinf #undef isfinite #include <SYCL/sycl.hpp> +#include <iostream> #include <map> #include <memory> #include <utility> diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorDeviceSycl.h b/unsupported/Eigen/CXX11/src/Tensor/TensorDeviceSycl.h index 25500683f..a78350bf3 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorDeviceSycl.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorDeviceSycl.h @@ -94,6 +94,10 @@ struct QueueInterface { } } + EIGEN_STRONG_INLINE void deallocate_all() const { + buffer_map.clear(); + } + EIGEN_STRONG_INLINE std::map<const uint8_t *, cl::sycl::buffer<uint8_t,1>>::iterator find_buffer(const void* ptr) const { auto it1 = buffer_map.find(static_cast<const uint8_t*>(ptr)); if (it1 != buffer_map.end()){ |