• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright © 2017 Google
3  * Copyright © 2019 Red Hat
4  *
5  * Permission is hereby granted, free of charge, to any person obtaining a
6  * copy of this software and associated documentation files (the "Software"),
7  * to deal in the Software without restriction, including without limitation
8  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
9  * and/or sell copies of the Software, and to permit persons to whom the
10  * Software is furnished to do so, subject to the following conditions:
11  *
12  * The above copyright notice and this permission notice (including the next
13  * paragraph) shall be included in all copies or substantial portions of the
14  * Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
19  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
21  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
22  * IN THE SOFTWARE.
23  */
24 
25 /* Rules for device selection.
26  * Is there an X or wayland connection open (or DISPLAY set).
27  * If no - try and find which device was the boot_vga device.
28  * If yes - try and work out which device is the connection primary,
29  * DRI_PRIME tagged overrides only work if bus info, =1 will just pick an alternate.
30  */
31 
32 #include <vulkan/vk_layer.h>
33 
34 #include <assert.h>
35 #include <stdio.h>
36 #include <string.h>
37 #include <fcntl.h>
38 #include <unistd.h>
39 
40 #include "device_select.h"
41 #include "c99_compat.h"
42 #include "hash_table.h"
43 #include "vk_util.h"
44 #include "c11/threads.h"
45 
46 struct instance_info {
47    PFN_vkDestroyInstance DestroyInstance;
48    PFN_vkEnumeratePhysicalDevices EnumeratePhysicalDevices;
49    PFN_vkEnumeratePhysicalDeviceGroups EnumeratePhysicalDeviceGroups;
50    PFN_vkGetInstanceProcAddr GetInstanceProcAddr;
51    PFN_GetPhysicalDeviceProcAddr  GetPhysicalDeviceProcAddr;
52    PFN_vkEnumerateDeviceExtensionProperties EnumerateDeviceExtensionProperties;
53    PFN_vkGetPhysicalDeviceProperties GetPhysicalDeviceProperties;
54    PFN_vkGetPhysicalDeviceProperties2 GetPhysicalDeviceProperties2;
55    bool has_pci_bus, has_vulkan11;
56    bool has_wayland, has_xcb;
57 };
58 
59 static struct hash_table *device_select_instance_ht = NULL;
60 static mtx_t device_select_mutex;
61 
62 static once_flag device_select_is_init = ONCE_FLAG_INIT;
63 
device_select_once_init(void)64 static void device_select_once_init(void) {
65    mtx_init(&device_select_mutex, mtx_plain);
66 }
67 
68 static void
device_select_init_instances(void)69 device_select_init_instances(void)
70 {
71    call_once(&device_select_is_init, device_select_once_init);
72 
73    mtx_lock(&device_select_mutex);
74    if (!device_select_instance_ht)
75       device_select_instance_ht = _mesa_hash_table_create(NULL, _mesa_hash_pointer,
76 							  _mesa_key_pointer_equal);
77    mtx_unlock(&device_select_mutex);
78 }
79 
80 static void
device_select_try_free_ht(void)81 device_select_try_free_ht(void)
82 {
83    mtx_lock(&device_select_mutex);
84    if (device_select_instance_ht) {
85       if (_mesa_hash_table_num_entries(device_select_instance_ht) == 0) {
86 	 _mesa_hash_table_destroy(device_select_instance_ht, NULL);
87 	 device_select_instance_ht = NULL;
88       }
89    }
90    mtx_unlock(&device_select_mutex);
91 }
92 
93 static void
device_select_layer_add_instance(VkInstance instance,struct instance_info * info)94 device_select_layer_add_instance(VkInstance instance, struct instance_info *info)
95 {
96    device_select_init_instances();
97    mtx_lock(&device_select_mutex);
98    _mesa_hash_table_insert(device_select_instance_ht, instance, info);
99    mtx_unlock(&device_select_mutex);
100 }
101 
102 static struct instance_info *
device_select_layer_get_instance(VkInstance instance)103 device_select_layer_get_instance(VkInstance instance)
104 {
105    struct hash_entry *entry;
106    struct instance_info *info = NULL;
107    mtx_lock(&device_select_mutex);
108    entry = _mesa_hash_table_search(device_select_instance_ht, (void *)instance);
109    if (entry)
110       info = (struct instance_info *)entry->data;
111    mtx_unlock(&device_select_mutex);
112    return info;
113 }
114 
115 static void
device_select_layer_remove_instance(VkInstance instance)116 device_select_layer_remove_instance(VkInstance instance)
117 {
118    mtx_lock(&device_select_mutex);
119    _mesa_hash_table_remove_key(device_select_instance_ht, instance);
120    mtx_unlock(&device_select_mutex);
121    device_select_try_free_ht();
122 }
123 
device_select_CreateInstance(const VkInstanceCreateInfo * pCreateInfo,const VkAllocationCallbacks * pAllocator,VkInstance * pInstance)124 static VkResult device_select_CreateInstance(const VkInstanceCreateInfo *pCreateInfo,
125 					     const VkAllocationCallbacks *pAllocator,
126 					     VkInstance *pInstance)
127 {
128    VkLayerInstanceCreateInfo *chain_info;
129    for(chain_info = (VkLayerInstanceCreateInfo*)pCreateInfo->pNext; chain_info; chain_info = (VkLayerInstanceCreateInfo*)chain_info->pNext)
130       if(chain_info->sType == VK_STRUCTURE_TYPE_LOADER_INSTANCE_CREATE_INFO && chain_info->function == VK_LAYER_LINK_INFO)
131          break;
132 
133    assert(chain_info->u.pLayerInfo);
134    struct instance_info *info = (struct instance_info *)calloc(1, sizeof(struct instance_info));
135 
136    info->GetInstanceProcAddr = chain_info->u.pLayerInfo->pfnNextGetInstanceProcAddr;
137    PFN_vkCreateInstance fpCreateInstance =
138       (PFN_vkCreateInstance)info->GetInstanceProcAddr(NULL, "vkCreateInstance");
139    if (fpCreateInstance == NULL) {
140       free(info);
141       return VK_ERROR_INITIALIZATION_FAILED;
142    }
143 
144    chain_info->u.pLayerInfo = chain_info->u.pLayerInfo->pNext;
145 
146    VkResult result = fpCreateInstance(pCreateInfo, pAllocator, pInstance);
147    if (result != VK_SUCCESS) {
148       free(info);
149       return result;
150    }
151 
152    for (unsigned i = 0; i < pCreateInfo->enabledExtensionCount; i++) {
153 #ifdef VK_USE_PLATFORM_WAYLAND_KHR
154       if (!strcmp(pCreateInfo->ppEnabledExtensionNames[i], VK_KHR_WAYLAND_SURFACE_EXTENSION_NAME))
155          info->has_wayland = true;
156 #endif
157 #ifdef VK_USE_PLATFORM_XCB_KHR
158       if (!strcmp(pCreateInfo->ppEnabledExtensionNames[i], VK_KHR_XCB_SURFACE_EXTENSION_NAME))
159          info->has_xcb = true;
160 #endif
161    }
162 
163    /*
164     * The loader is currently not able to handle GetPhysicalDeviceProperties2KHR calls in
165     * EnumeratePhysicalDevices when there are other layers present. To avoid mysterious crashes
166     * for users just use only the vulkan version for now.
167     */
168    info->has_vulkan11 = pCreateInfo->pApplicationInfo &&
169                         pCreateInfo->pApplicationInfo->apiVersion >= VK_MAKE_VERSION(1, 1, 0);
170 
171    info->GetPhysicalDeviceProcAddr = (PFN_GetPhysicalDeviceProcAddr)info->GetInstanceProcAddr(*pInstance, "vk_layerGetPhysicalDeviceProcAddr");
172 #define DEVSEL_GET_CB(func) info->func = (PFN_vk##func)info->GetInstanceProcAddr(*pInstance, "vk" #func)
173    DEVSEL_GET_CB(DestroyInstance);
174    DEVSEL_GET_CB(EnumeratePhysicalDevices);
175    DEVSEL_GET_CB(EnumeratePhysicalDeviceGroups);
176    DEVSEL_GET_CB(GetPhysicalDeviceProperties);
177    DEVSEL_GET_CB(EnumerateDeviceExtensionProperties);
178    if (info->has_vulkan11)
179       DEVSEL_GET_CB(GetPhysicalDeviceProperties2);
180 #undef DEVSEL_GET_CB
181 
182    device_select_layer_add_instance(*pInstance, info);
183 
184    return VK_SUCCESS;
185 }
186 
device_select_DestroyInstance(VkInstance instance,const VkAllocationCallbacks * pAllocator)187 static void device_select_DestroyInstance(VkInstance instance, const VkAllocationCallbacks* pAllocator)
188 {
189    struct instance_info *info = device_select_layer_get_instance(instance);
190 
191    device_select_layer_remove_instance(instance);
192    info->DestroyInstance(instance, pAllocator);
193    free(info);
194 }
195 
get_device_properties(const struct instance_info * info,VkPhysicalDevice device,VkPhysicalDeviceProperties2 * properties)196 static void get_device_properties(const struct instance_info *info, VkPhysicalDevice device, VkPhysicalDeviceProperties2 *properties)
197 {
198     info->GetPhysicalDeviceProperties(device, &properties->properties);
199 
200     if (info->GetPhysicalDeviceProperties2 && properties->properties.apiVersion >= VK_API_VERSION_1_1)
201         info->GetPhysicalDeviceProperties2(device, properties);
202 }
203 
print_gpu(const struct instance_info * info,unsigned index,VkPhysicalDevice device)204 static void print_gpu(const struct instance_info *info, unsigned index, VkPhysicalDevice device)
205 {
206    const char *type = "";
207    VkPhysicalDevicePCIBusInfoPropertiesEXT ext_pci_properties = (VkPhysicalDevicePCIBusInfoPropertiesEXT) {
208       .sType = VK_STRUCTURE_TYPE_PHYSICAL_DEVICE_PCI_BUS_INFO_PROPERTIES_EXT
209    };
210    VkPhysicalDeviceProperties2KHR properties = (VkPhysicalDeviceProperties2KHR){
211       .sType = VK_STRUCTURE_TYPE_PHYSICAL_DEVICE_PROPERTIES_2_KHR
212    };
213    if (info->has_vulkan11 && info->has_pci_bus)
214       properties.pNext = &ext_pci_properties;
215    get_device_properties(info, device, &properties);
216 
217    switch(properties.properties.deviceType) {
218    case VK_PHYSICAL_DEVICE_TYPE_OTHER:
219    default:
220       type = "other";
221       break;
222    case VK_PHYSICAL_DEVICE_TYPE_INTEGRATED_GPU:
223       type = "integrated GPU";
224       break;
225    case VK_PHYSICAL_DEVICE_TYPE_DISCRETE_GPU:
226       type = "discrete GPU";
227       break;
228    case VK_PHYSICAL_DEVICE_TYPE_VIRTUAL_GPU:
229       type = "virtual GPU";
230       break;
231    case VK_PHYSICAL_DEVICE_TYPE_CPU:
232       type = "CPU";
233       break;
234    }
235    fprintf(stderr, "  GPU %d: %x:%x \"%s\" %s", index, properties.properties.vendorID,
236            properties.properties.deviceID, properties.properties.deviceName, type);
237    if (info->has_pci_bus)
238       fprintf(stderr, " %04x:%02x:%02x.%x", ext_pci_properties.pciDomain,
239               ext_pci_properties.pciBus, ext_pci_properties.pciDevice,
240               ext_pci_properties.pciFunction);
241    fprintf(stderr, "\n");
242 }
243 
fill_drm_device_info(const struct instance_info * info,struct device_pci_info * drm_device,VkPhysicalDevice device)244 static bool fill_drm_device_info(const struct instance_info *info,
245                                  struct device_pci_info *drm_device,
246                                  VkPhysicalDevice device)
247 {
248    VkPhysicalDevicePCIBusInfoPropertiesEXT ext_pci_properties = (VkPhysicalDevicePCIBusInfoPropertiesEXT) {
249       .sType = VK_STRUCTURE_TYPE_PHYSICAL_DEVICE_PCI_BUS_INFO_PROPERTIES_EXT
250    };
251 
252    VkPhysicalDeviceProperties2KHR properties = (VkPhysicalDeviceProperties2KHR){
253       .sType = VK_STRUCTURE_TYPE_PHYSICAL_DEVICE_PROPERTIES_2_KHR
254    };
255 
256    if (info->has_vulkan11 && info->has_pci_bus)
257       properties.pNext = &ext_pci_properties;
258    get_device_properties(info, device, &properties);
259 
260    drm_device->cpu_device = properties.properties.deviceType == VK_PHYSICAL_DEVICE_TYPE_CPU;
261    drm_device->dev_info.vendor_id = properties.properties.vendorID;
262    drm_device->dev_info.device_id = properties.properties.deviceID;
263    if (info->has_vulkan11 && info->has_pci_bus) {
264      drm_device->has_bus_info = true;
265      drm_device->bus_info.domain = ext_pci_properties.pciDomain;
266      drm_device->bus_info.bus = ext_pci_properties.pciBus;
267      drm_device->bus_info.dev = ext_pci_properties.pciDevice;
268      drm_device->bus_info.func = ext_pci_properties.pciFunction;
269    }
270    return drm_device->cpu_device;
271 }
272 
device_select_find_explicit_default(struct device_pci_info * pci_infos,uint32_t device_count,const char * selection)273 static int device_select_find_explicit_default(struct device_pci_info *pci_infos,
274                                                uint32_t device_count,
275                                                const char *selection)
276 {
277    int default_idx = -1;
278    unsigned vendor_id, device_id;
279    int matched = sscanf(selection, "%x:%x", &vendor_id, &device_id);
280    if (matched != 2)
281       return default_idx;
282 
283    for (unsigned i = 0; i < device_count; ++i) {
284       if (pci_infos[i].dev_info.vendor_id == vendor_id &&
285           pci_infos[i].dev_info.device_id == device_id)
286          default_idx = i;
287    }
288    return default_idx;
289 }
290 
device_select_find_dri_prime_tag_default(struct device_pci_info * pci_infos,uint32_t device_count,const char * dri_prime)291 static int device_select_find_dri_prime_tag_default(struct device_pci_info *pci_infos,
292                                                     uint32_t device_count,
293                                                     const char *dri_prime)
294 {
295    int default_idx = -1;
296    for (unsigned i = 0; i < device_count; ++i) {
297       char *tag = NULL;
298       if (asprintf(&tag, "pci-%04x_%02x_%02x_%1u",
299                    pci_infos[i].bus_info.domain,
300                    pci_infos[i].bus_info.bus,
301                    pci_infos[i].bus_info.dev,
302                    pci_infos[i].bus_info.func) >= 0) {
303          if (strcmp(dri_prime, tag))
304             default_idx = i;
305       }
306       free(tag);
307    }
308    return default_idx;
309 }
310 
device_select_find_boot_vga_default(struct device_pci_info * pci_infos,uint32_t device_count)311 static int device_select_find_boot_vga_default(struct device_pci_info *pci_infos,
312                                                uint32_t device_count)
313 {
314    char boot_vga_path[1024];
315    int default_idx = -1;
316    for (unsigned i = 0; i < device_count; ++i) {
317       /* fallback to probing the pci bus boot_vga device. */
318       snprintf(boot_vga_path, 1023, "/sys/bus/pci/devices/%04x:%02x:%02x.%x/boot_vga", pci_infos[i].bus_info.domain,
319                pci_infos[i].bus_info.bus, pci_infos[i].bus_info.dev, pci_infos[i].bus_info.func);
320       int fd = open(boot_vga_path, O_RDONLY);
321       if (fd != -1) {
322          uint8_t val;
323          if (read(fd, &val, 1) == 1) {
324             if (val == '1')
325                default_idx = i;
326          }
327          close(fd);
328       }
329       if (default_idx != -1)
330          break;
331    }
332    return default_idx;
333 }
334 
device_select_find_non_cpu(struct device_pci_info * pci_infos,uint32_t device_count)335 static int device_select_find_non_cpu(struct device_pci_info *pci_infos,
336                                       uint32_t device_count)
337 {
338    int default_idx = -1;
339 
340    /* pick first GPU device */
341    for (unsigned i = 0; i < device_count; ++i) {
342       if (!pci_infos[i].cpu_device){
343          default_idx = i;
344          break;
345       }
346    }
347    return default_idx;
348 }
349 
find_non_cpu_skip(struct device_pci_info * pci_infos,uint32_t device_count,int skip_idx)350 static int find_non_cpu_skip(struct device_pci_info *pci_infos,
351                         uint32_t device_count,
352                         int skip_idx)
353 {
354    for (unsigned i = 0; i < device_count; ++i) {
355       if (i == skip_idx)
356          continue;
357       if (pci_infos[i].cpu_device)
358          continue;
359       return i;
360    }
361    return -1;
362 }
363 
get_default_device(const struct instance_info * info,const char * selection,uint32_t physical_device_count,VkPhysicalDevice * pPhysicalDevices)364 static uint32_t get_default_device(const struct instance_info *info,
365                                    const char *selection,
366                                    uint32_t physical_device_count,
367                                    VkPhysicalDevice *pPhysicalDevices)
368 {
369    int default_idx = -1;
370    const char *dri_prime = getenv("DRI_PRIME");
371    bool dri_prime_is_one = false;
372    int cpu_count = 0;
373    if (dri_prime && !strcmp(dri_prime, "1"))
374       dri_prime_is_one = true;
375 
376    if (dri_prime && !dri_prime_is_one && !info->has_pci_bus) {
377       fprintf(stderr, "device-select: cannot correctly use DRI_PRIME tag\n");
378    }
379 
380    struct device_pci_info *pci_infos = (struct device_pci_info *)calloc(physical_device_count, sizeof(struct device_pci_info));
381    if (!pci_infos)
382      return 0;
383 
384    for (unsigned i = 0; i < physical_device_count; ++i) {
385       cpu_count += fill_drm_device_info(info, &pci_infos[i], pPhysicalDevices[i]) ? 1 : 0;
386    }
387 
388    if (selection)
389       default_idx = device_select_find_explicit_default(pci_infos, physical_device_count, selection);
390    if (default_idx == -1 && info->has_pci_bus && dri_prime && !dri_prime_is_one)
391       default_idx = device_select_find_dri_prime_tag_default(pci_infos, physical_device_count, dri_prime);
392    if (default_idx == -1 && info->has_wayland)
393       default_idx = device_select_find_wayland_pci_default(pci_infos, physical_device_count);
394    if (default_idx == -1 && info->has_xcb)
395       default_idx = device_select_find_xcb_pci_default(pci_infos, physical_device_count);
396    if (default_idx == -1 && info->has_pci_bus)
397       default_idx = device_select_find_boot_vga_default(pci_infos, physical_device_count);
398    if (default_idx == -1 && cpu_count)
399       default_idx = device_select_find_non_cpu(pci_infos, physical_device_count);
400 
401    /* DRI_PRIME=1 handling - pick any other device than default. */
402    if (default_idx != -1 && dri_prime_is_one && physical_device_count > (cpu_count + 1)) {
403       if (default_idx == 0 || default_idx == 1)
404          default_idx = find_non_cpu_skip(pci_infos, physical_device_count, default_idx);
405    }
406    free(pci_infos);
407    return default_idx == -1 ? 0 : default_idx;
408 }
409 
device_select_EnumeratePhysicalDevices(VkInstance instance,uint32_t * pPhysicalDeviceCount,VkPhysicalDevice * pPhysicalDevices)410 static VkResult device_select_EnumeratePhysicalDevices(VkInstance instance,
411 						       uint32_t* pPhysicalDeviceCount,
412 						       VkPhysicalDevice *pPhysicalDevices)
413 {
414    struct instance_info *info = device_select_layer_get_instance(instance);
415    uint32_t physical_device_count = 0;
416    uint32_t selected_physical_device_count = 0;
417    const char* selection = getenv("MESA_VK_DEVICE_SELECT");
418    VkResult result = info->EnumeratePhysicalDevices(instance, &physical_device_count, NULL);
419    VK_OUTARRAY_MAKE(out, pPhysicalDevices, pPhysicalDeviceCount);
420    if (result != VK_SUCCESS)
421       return result;
422 
423    VkPhysicalDevice *physical_devices = (VkPhysicalDevice*)calloc(sizeof(VkPhysicalDevice),  physical_device_count);
424    VkPhysicalDevice *selected_physical_devices = (VkPhysicalDevice*)calloc(sizeof(VkPhysicalDevice),
425                                                                            physical_device_count);
426 
427    if (!physical_devices || !selected_physical_devices) {
428       result = VK_ERROR_OUT_OF_HOST_MEMORY;
429       goto out;
430    }
431 
432    result = info->EnumeratePhysicalDevices(instance, &physical_device_count, physical_devices);
433    if (result != VK_SUCCESS)
434       goto out;
435 
436    for (unsigned i = 0; i < physical_device_count; i++) {
437       uint32_t count;
438       info->EnumerateDeviceExtensionProperties(physical_devices[i], NULL, &count, NULL);
439       if (count > 0) {
440 	 VkExtensionProperties *extensions = calloc(count, sizeof(VkExtensionProperties));
441          if (info->EnumerateDeviceExtensionProperties(physical_devices[i], NULL, &count, extensions) == VK_SUCCESS) {
442 	    for (unsigned j = 0; j < count; j++) {
443                if (!strcmp(extensions[j].extensionName, VK_EXT_PCI_BUS_INFO_EXTENSION_NAME))
444                   info->has_pci_bus = true;
445             }
446          }
447 	 free(extensions);
448       }
449    }
450    if (selection && strcmp(selection, "list") == 0) {
451       fprintf(stderr, "selectable devices:\n");
452       for (unsigned i = 0; i < physical_device_count; ++i)
453          print_gpu(info, i, physical_devices[i]);
454       exit(0);
455    } else {
456       unsigned selected_index = get_default_device(info, selection, physical_device_count, physical_devices);
457       selected_physical_device_count = physical_device_count;
458       selected_physical_devices[0] = physical_devices[selected_index];
459       for (unsigned i = 0; i < physical_device_count - 1; ++i) {
460          unsigned  this_idx = i < selected_index ? i : i + 1;
461          selected_physical_devices[i + 1] = physical_devices[this_idx];
462       }
463    }
464 
465    if (selected_physical_device_count == 0) {
466       fprintf(stderr, "WARNING: selected no devices with MESA_VK_DEVICE_SELECT\n");
467    }
468 
469    assert(result == VK_SUCCESS);
470 
471    for (unsigned i = 0; i < selected_physical_device_count; i++) {
472       vk_outarray_append(&out, ent) {
473          *ent = selected_physical_devices[i];
474       }
475    }
476    result = vk_outarray_status(&out);
477  out:
478    free(physical_devices);
479    free(selected_physical_devices);
480    return result;
481 }
482 
device_select_EnumeratePhysicalDeviceGroups(VkInstance instance,uint32_t * pPhysicalDeviceGroupCount,VkPhysicalDeviceGroupProperties * pPhysicalDeviceGroups)483 static VkResult device_select_EnumeratePhysicalDeviceGroups(VkInstance instance,
484                                                             uint32_t* pPhysicalDeviceGroupCount,
485                                                             VkPhysicalDeviceGroupProperties *pPhysicalDeviceGroups)
486 {
487    struct instance_info *info = device_select_layer_get_instance(instance);
488    uint32_t physical_device_group_count = 0;
489    uint32_t selected_physical_device_group_count = 0;
490    VkResult result = info->EnumeratePhysicalDeviceGroups(instance, &physical_device_group_count, NULL);
491    VK_OUTARRAY_MAKE(out, pPhysicalDeviceGroups, pPhysicalDeviceGroupCount);
492 
493    if (result != VK_SUCCESS)
494       return result;
495 
496    VkPhysicalDeviceGroupProperties *physical_device_groups = (VkPhysicalDeviceGroupProperties*)calloc(sizeof(VkPhysicalDeviceGroupProperties), physical_device_group_count);
497    VkPhysicalDeviceGroupProperties *selected_physical_device_groups = (VkPhysicalDeviceGroupProperties*)calloc(sizeof(VkPhysicalDeviceGroupProperties), physical_device_group_count);
498 
499    if (!physical_device_groups || !selected_physical_device_groups) {
500       result = VK_ERROR_OUT_OF_HOST_MEMORY;
501       goto out;
502    }
503 
504    result = info->EnumeratePhysicalDeviceGroups(instance, &physical_device_group_count, physical_device_groups);
505    if (result != VK_SUCCESS)
506       goto out;
507 
508    /* just sort groups with CPU devices to the end? - assume nobody will mix these */
509    int num_gpu_groups = 0;
510    int num_cpu_groups = 0;
511    selected_physical_device_group_count = physical_device_group_count;
512    for (unsigned i = 0; i < physical_device_group_count; i++) {
513       bool group_has_cpu_device = false;
514       for (unsigned j = 0; j < physical_device_groups[i].physicalDeviceCount; j++) {
515          VkPhysicalDevice physical_device = physical_device_groups[i].physicalDevices[j];
516          VkPhysicalDeviceProperties2KHR properties = (VkPhysicalDeviceProperties2KHR){
517             .sType = VK_STRUCTURE_TYPE_PHYSICAL_DEVICE_PROPERTIES_2_KHR
518          };
519          info->GetPhysicalDeviceProperties(physical_device, &properties.properties);
520          group_has_cpu_device = properties.properties.deviceType == VK_PHYSICAL_DEVICE_TYPE_CPU;
521       }
522 
523       if (group_has_cpu_device) {
524          selected_physical_device_groups[physical_device_group_count - num_cpu_groups - 1] = physical_device_groups[i];
525          num_cpu_groups++;
526       } else {
527          selected_physical_device_groups[num_gpu_groups] = physical_device_groups[i];
528          num_gpu_groups++;
529       }
530    }
531 
532    assert(result == VK_SUCCESS);
533 
534    for (unsigned i = 0; i < selected_physical_device_group_count; i++) {
535       vk_outarray_append(&out, ent) {
536          *ent = selected_physical_device_groups[i];
537       }
538    }
539    result = vk_outarray_status(&out);
540 out:
541    free(physical_device_groups);
542    free(selected_physical_device_groups);
543    return result;
544 }
545 
get_pdevice_proc_addr(VkInstance instance,const char * name)546 static void  (*get_pdevice_proc_addr(VkInstance instance, const char* name))()
547 {
548    struct instance_info *info = device_select_layer_get_instance(instance);
549    return info->GetPhysicalDeviceProcAddr(instance, name);
550 }
551 
get_instance_proc_addr(VkInstance instance,const char * name)552 static void  (*get_instance_proc_addr(VkInstance instance, const char* name))()
553 {
554    if (strcmp(name, "vkGetInstanceProcAddr") == 0)
555       return (void(*)())get_instance_proc_addr;
556    if (strcmp(name, "vkCreateInstance") == 0)
557       return (void(*)())device_select_CreateInstance;
558    if (strcmp(name, "vkDestroyInstance") == 0)
559       return (void(*)())device_select_DestroyInstance;
560    if (strcmp(name, "vkEnumeratePhysicalDevices") == 0)
561       return (void(*)())device_select_EnumeratePhysicalDevices;
562    if (strcmp(name, "vkEnumeratePhysicalDeviceGroups") == 0)
563       return (void(*)())device_select_EnumeratePhysicalDeviceGroups;
564 
565    struct instance_info *info = device_select_layer_get_instance(instance);
566    return info->GetInstanceProcAddr(instance, name);
567 }
568 
vkNegotiateLoaderLayerInterfaceVersion(VkNegotiateLayerInterface * pVersionStruct)569 VK_LAYER_EXPORT VkResult vkNegotiateLoaderLayerInterfaceVersion(VkNegotiateLayerInterface *pVersionStruct)
570 {
571    if (pVersionStruct->loaderLayerInterfaceVersion < 2)
572       return VK_ERROR_INITIALIZATION_FAILED;
573    pVersionStruct->loaderLayerInterfaceVersion = 2;
574 
575    pVersionStruct->pfnGetInstanceProcAddr = get_instance_proc_addr;
576    pVersionStruct->pfnGetPhysicalDeviceProcAddr = get_pdevice_proc_addr;
577 
578    return VK_SUCCESS;
579 }
580