Mesa (master): clover: query driver for the max number of compute units

[Tom Stellard]

Module: Mesa
Branch: master
Commit: 2a0dffa0c91cc2277f82c2b87277f56e525f87ab
URL:    http://cgit.freedesktop.org/mesa/mesa/commit/?id=2a0dffa0c91cc2277f82c2b87277f56e525f87ab

Author: Bruno Jiménez <brunojimen at gmail.com>
Date:   Fri May 30 17:31:12 2014 +0200

clover: query driver for the max number of compute units

Reviewed-by: Tom Stellard <thomas.stellard at amd.com>
Reviewed-by: Francisco Jerez <currojerez at riseup.net>

---

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

diff --git a/src/gallium/state_trackers/clover/api/device.cpp b/src/gallium/state_trackers/clover/api/device.cpp
index 1bc2692..97b2cf9 100644
--- a/src/gallium/state_trackers/clover/api/device.cpp
+++ b/src/gallium/state_trackers/clover/api/device.cpp
@@ -109,7 +109,7 @@ clGetDeviceInfo(cl_device_id d_dev, cl_device_info param,
       break;
 
    case CL_DEVICE_MAX_COMPUTE_UNITS:
-      buf.as_scalar<cl_uint>() = 1;
+      buf.as_scalar<cl_uint>() = dev.max_compute_units();
       break;
 
    case CL_DEVICE_MAX_WORK_ITEM_DIMENSIONS:
diff --git a/src/gallium/state_trackers/clover/core/device.cpp b/src/gallium/state_trackers/clover/core/device.cpp
index bc3e3e6..b6078db 100644
--- a/src/gallium/state_trackers/clover/core/device.cpp
+++ b/src/gallium/state_trackers/clover/core/device.cpp
@@ -163,6 +163,12 @@ device::max_clock_frequency() const {
                                       PIPE_COMPUTE_CAP_MAX_CLOCK_FREQUENCY)[0];
 }
 
+cl_uint
+device::max_compute_units() const {
+   return get_compute_param<uint32_t>(pipe,
+                                      PIPE_COMPUTE_CAP_MAX_COMPUTE_UNITS)[0];
+}
+
 std::vector<size_t>
 device::max_block_size() const {
    auto v = get_compute_param<uint64_t>(pipe, PIPE_COMPUTE_CAP_MAX_BLOCK_SIZE);
diff --git a/src/gallium/state_trackers/clover/core/device.hpp b/src/gallium/state_trackers/clover/core/device.hpp
index 3662c6b..731c31e 100644
--- a/src/gallium/state_trackers/clover/core/device.hpp
+++ b/src/gallium/state_trackers/clover/core/device.hpp
@@ -62,6 +62,7 @@ namespace clover {
       size_t max_threads_per_block() const;
       cl_ulong max_mem_alloc_size() const;
       cl_uint max_clock_frequency() const;
+      cl_uint max_compute_units() const;
 
       std::vector<size_t> max_block_size() const;
       std::string device_name() const;