devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Distinguishing between internal memory allocation/deallocation from explicit user memory allocation/deallocation.

Browse Source
This commit is contained in:
Mehdi Goli
2018-08-01 11:56:30 +01:00
parent edf46bd7a2
commit d7a8414848
6 changed files with 34 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
unsupported/Eigen/CXX11/src/Tensor/TensorDeviceGpu.h
View File

@@ -159,6 +159,13 @@ class GpuStreamDevice : public StreamInterface {
err = gpuFree(buffer);
gpu_assert(err == gpuSuccess);
}
EIGEN_STRONG_INLINE void* allocate_temp(size_t num_bytes) const {
return stream_->allocate(num_bytes);
}
EIGEN_STRONG_INLINE void deallocate_temp(void* buffer) const {
stream_->deallocate(buffer);
}
virtual void* scratchpad() const {
if (scratch_ == NULL) {