From f053edf6b83e5b0727dc9420bd5d7c61acf4de6e Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 19 Oct 2020 13:14:45 -0500
Subject: [PATCH] ImmediateAllocator: use clEnqueueMigrateMemObjects to force
allocation on CL1.2+ devices
---
src/wrap_mempool.cpp | 27 +++++++++++++++++++--------
1 file changed, 19 insertions(+), 8 deletions(-)
diff --git a/src/wrap_mempool.cpp b/src/wrap_mempool.cpp
index 3f26a2f9..4ccb61f2 100644
--- a/src/wrap_mempool.cpp
+++ b/src/wrap_mempool.cpp
@@ -154,14 +154,25 @@ namespace
// reported in a deferred manner, it has no way to react
// (e.g. by freeing unused memory) because it is not part of
// the call stack.)
- unsigned zero = 0;
- PYOPENCL_CALL_GUARDED(clEnqueueWriteBuffer, (
- m_queue.data(),
- ptr,
- /* is blocking */ CL_FALSE,
- 0, std::min(s, sizeof(zero)), &zero,
- 0, NULL, NULL
- ));
+ if (m_queue.get_hex_device_version() < 0x1020)
+ {
+ unsigned zero = 0;
+ PYOPENCL_CALL_GUARDED(clEnqueueWriteBuffer, (
+ m_queue.data(),
+ ptr,
+ /* is blocking */ CL_FALSE,
+ 0, std::min(s, sizeof(zero)), &zero,
+ 0, NULL, NULL
+ ));
+ }
+ else
+ {
+ PYOPENCL_CALL_GUARDED(clEnqueueMigrateMemObjects, (
+ m_queue.data(),
+ 1, &ptr, CL_MIGRATE_MEM_OBJECT_CONTENT_UNDEFINED,
+ 0, NULL, NULL
+ ));
+ }
// No need to wait for completion here. clWaitForEvents (e.g.)
// cannot return mem object allocation failures. This implies that
--
GitLab