diff --git a/pyopencl/characterize/__init__.py b/pyopencl/characterize/__init__.py index 835cae09623e2cb12b912279373706e8455fd3ca..238bfe5fd3b4e0dcc95ea07f64b47ec2d211d45f 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)