Correct compute::context::get_devices method

get a vector of cl_device_id and then use the device ids to populate a vector of compute::device Signed-off-by: Mateusz Jablonski <[email protected]>
boostorg · Jul 8, 2020 · 881ca42 · 881ca42
1 parent 36c8913
commit 881ca42
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/include/boost/compute/context.hpp b/include/boost/compute/context.hpp
@@ -187,7 +187,15 @@ class context
     /// Returns a vector of devices for the context.
     std::vector<device> get_devices() const
     {
-        return get_info<std::vector<device> >(CL_CONTEXT_DEVICES);
+        std::vector<cl_device_id> device_ids =
+            get_info<std::vector<cl_device_id> >(CL_CONTEXT_DEVICES);
+
+        std::vector<device> devices;
+        for(size_t i = 0; i < device_ids.size(); i++) {
+            devices.push_back(device(device_ids[i]));
+        }
+
+        return devices;
     }
 
     /// Returns information about the context.