COMPMID-337: Adding OpenCL SVM support.
Change-Id: I250d6a1daeccf91d97b6da65aec53b02cf6046a7
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/116140
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
diff --git a/arm_compute/runtime/CL/CLTensorAllocator.h b/arm_compute/runtime/CL/CLTensorAllocator.h
index 7515074..6929d55 100644
--- a/arm_compute/runtime/CL/CLTensorAllocator.h
+++ b/arm_compute/runtime/CL/CLTensorAllocator.h
@@ -27,7 +27,7 @@
#include "arm_compute/runtime/ITensorAllocator.h"
#include "arm_compute/core/CL/OpenCL.h"
-
+#include "arm_compute/runtime/CL/SVMMemory.h"
#include <cstdint>
namespace arm_compute
@@ -68,6 +68,9 @@
* @return pointer to the CL data.
*/
const cl::Buffer &cl_data() const;
+ /** SVM memory */
+ void *svm_ptr();
+
/** Enqueue a map operation of the allocated buffer on the given queue.
*
* @param[in,out] q The CL command queue to use for the mapping operation.
@@ -121,6 +124,7 @@
cl::Buffer _buffer; /**< OpenCL buffer containing the tensor data. */
uint8_t *_mapping; /**< Pointer to the CPU mapping of the OpenCL buffer. */
CLTensor *_owner; /**< Owner of the allocator */
+ SVMMemory _svm_memory; /**< Svm memory */
};
}
#endif /* __ARM_COMPUTE_CLTENSORALLOCATOR_H__ */