From 4f41f14b1d32d7e768af1690662efec8ebadf4e2 Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Wed, 22 Jun 2022 09:37:41 -0500
Subject: [PATCH] SVM characterize: add atomics (#577)
---
pyopencl/characterize/__init__.py | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/pyopencl/characterize/__init__.py b/pyopencl/characterize/__init__.py
index 835cae09..238bfe5f 100644
--- a/pyopencl/characterize/__init__.py
+++ b/pyopencl/characterize/__init__.py
@@ -417,3 +417,13 @@ def has_fine_grain_system_svm(dev):
return (_may_have_svm(dev)
and bool(dev.svm_capabilities
& cl.device_svm_capabilities.FINE_GRAIN_SYSTEM))
+
+
+def has_fine_grain_buffer_svm_atomics(dev):
+ return has_fine_grain_buffer_svm(dev) and bool(dev.svm_capabilities
+ & cl.device_svm_capabilities.ATOMICS)
+
+
+def has_fine_grain_system_svm_atomics(dev):
+ return has_fine_grain_system_svm(dev) and bool(dev.svm_capabilities
+ & cl.device_svm_capabilities.ATOMICS)
--
GitLab