[Mesa-dev] [PATCH 2/2] clover: Query and export int64 atomics

[Jan Vesely]

Signed-off-by: Jan Vesely <jan.vesely at rutgers.edu>
---
CC: Francisco Jerez <currojerez at riseup.net>
This exposes both extensions on GCN+ GPUs.

 src/gallium/state_trackers/clover/api/device.cpp  | 6 ++++--
 src/gallium/state_trackers/clover/core/device.cpp | 6 ++++++
 src/gallium/state_trackers/clover/core/device.hpp | 1 +
 3 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/gallium/state_trackers/clover/api/device.cpp b/src/gallium/state_trackers/clover/api/device.cpp
index 26e88aadc7..3572bb0c92 100644
--- a/src/gallium/state_trackers/clover/api/device.cpp
+++ b/src/gallium/state_trackers/clover/api/device.cpp
@@ -331,11 +331,13 @@ clGetDeviceInfo(cl_device_id d_dev, cl_device_info param,
 
    case CL_DEVICE_EXTENSIONS:
       buf.as_string() =
-         "cl_khr_global_int32_base_atomics"
+         "cl_khr_byte_addressable_store"
+         " cl_khr_global_int32_base_atomics"
          " cl_khr_global_int32_extended_atomics"
          " cl_khr_local_int32_base_atomics"
          " cl_khr_local_int32_extended_atomics"
-         " cl_khr_byte_addressable_store"
+         + std::string(dev.has_int64_atomics() ? " cl_khr_int64_base_atomics" : "")
+         + std::string(dev.has_int64_atomics() ? " cl_khr_int64_extended_atomics" : "")
          + std::string(dev.has_doubles() ? " cl_khr_fp64" : "")
          + std::string(dev.has_halves() ? " cl_khr_fp16" : "");
       break;
diff --git a/src/gallium/state_trackers/clover/core/device.cpp b/src/gallium/state_trackers/clover/core/device.cpp
index c0ee0acaaf..9dd7eed3f1 100644
--- a/src/gallium/state_trackers/clover/core/device.cpp
+++ b/src/gallium/state_trackers/clover/core/device.cpp
@@ -197,6 +197,12 @@ device::has_halves() const {
 }
 
 bool
+device::has_int64_atomics() const {
+   return pipe->get_shader_param(pipe, PIPE_SHADER_COMPUTE,
+                                 PIPE_SHADER_CAP_INT64_ATOMICS);
+}
+
+bool
 device::has_unified_memory() const {
    return pipe->get_param(pipe, PIPE_CAP_UMA);
 }
diff --git a/src/gallium/state_trackers/clover/core/device.hpp b/src/gallium/state_trackers/clover/core/device.hpp
index c61be790a9..85cd031676 100644
--- a/src/gallium/state_trackers/clover/core/device.hpp
+++ b/src/gallium/state_trackers/clover/core/device.hpp
@@ -68,6 +68,7 @@ namespace clover {
       bool image_support() const;
       bool has_doubles() const;
       bool has_halves() const;
+      bool has_int64_atomics() const;
       bool has_unified_memory() const;
       cl_uint mem_base_addr_align() const;
 
-- 
2.13.5