srkreddy1238 commented on code in PR #15058:
URL: https://github.com/apache/tvm/pull/15058#discussion_r1349641473
##########
src/runtime/opencl/opencl_device_api.cc:
##########
@@ -222,8 +227,61 @@ void*
OpenCLWorkspace::CreateHostPtrIfEnabled(cl::BufferDescriptor* desc, Device
void* OpenCLWorkspace::AllocDataSpace(Device dev, size_t size, size_t
alignment,
DLDataType type_hint) {
this->Init();
+ return AllocCLBuffer(dev, size, alignment, type_hint);
+}
+
+void* OpenCLWorkspace::AllocDataSpace(Device dev, size_t width, size_t height,
DLDataType type_hint,
+ Optional<String> mem_scope) {
+ // Texture allocation given width and height
+ cl_uint row_align = GetImageAlignment(dev.device_id);
+ size_t pixel_size = (type_hint.bits * type_hint.lanes + 7) / 8;
+ size_t row_pitch = ALIGN_UP(width * pixel_size * 4, row_align); // CL_RGBA
= 4
+ size_t mem_size = row_pitch * height;
+
+ // Alloc back buffer from pool
+ cl::BufferDescriptor* back_buffer = nullptr;
+ if (IsBufferToImageSupported(dev.device_id)) {
+ auto buf = MemoryManager::GetOrCreateAllocator(dev, AllocatorType::kPooled)
Review Comment:
Now the device API follows the executor allocator type.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
