如何利用Metal的registryID属性匹配IOPCIDevice获取的同一GPU？

Hey there, let's work through this GPU matching problem together. I've tackled similar tasks before when bridging IOKit and Metal data, so here's a solid approach that works even for older Mac Pros:

Core Concept: Matching registryID Correctly #

First, a critical point: Metal's MTLDevice.registryID (added in Metal 2, macOS 10.13+) directly corresponds to the IOPCIDevice's registry entry ID, which you retrieve via the kIORegistryEntryIDKey property—not the raw iterator result. That's likely where your initial matching failed.

Step 1: Get an Iterator for All PCIe GPU Devices #

First, we need to filter IOKit devices to only grab GPU hardware (using the PCI device class code for display controllers):

#import <IOKit/pci/IOPCIDevice.h>
#import <Metal/Metal.h>

io_iterator_t CreateGPUIterator(void) {
    // Start with a matching dictionary for PCI devices
    CFMutableDictionaryRef matchingDict = IOServiceMatching(kIOPCIDeviceClass);
    if (!matchingDict) return IO_OBJECT_NULL;
    
    // Filter for display controllers (PCI class code 0x030000)
    CFDictionarySetValue(matchingDict, CFSTR(kIOPCIExpressPrimaryClassKey), 
                         [NSNumber numberWithUnsignedInt:0x030000]);
    
    io_iterator_t iterator;
    kern_return_t kr = IOServiceGetMatchingServices(kIOMasterPortDefault, matchingDict, &iterator);
    return (kr == KERN_SUCCESS) ? iterator : IO_OBJECT_NULL;
}

Step 2: Extract the Registry ID from an IOPCIDevice #

This helper pulls the correct ID that matches Metal's registryID:

uint64_t GetRegistryIDForIOPCIDevice(io_service_t device) {
    uint64_t registryID = 0;
    CFNumberRef idProperty = (CFNumberRef)IORegistryEntryCreateCFProperty(
        device, CFSTR(kIORegistryEntryIDKey), kCFAllocatorDefault, 0);
    
    if (idProperty) {
        CFNumberGetValue(idProperty, kCFNumberSInt64Type, &registryID);
        CFRelease(idProperty);
    }
    return registryID;
}

Step 3: Match IOKit GPUs to Metal Devices #

Now we'll iterate through both sets of devices, with a fallback for older systems/Metal 1 devices (like pre-2014 Mac Pros) that don't have registryID:

void MatchGPUsWithMetalSupport(void) {
    // Grab all Metal-compatible devices first
    NSArray<id<MTLDevice>> *metalDevices = MTLCopyAllDevices();
    if (!metalDevices.count) {
        NSLog(@"No Metal-compatible GPUs found");
        return;
    }
    
    io_iterator_t gpuIterator = CreateGPUIterator();
    if (gpuIterator == IO_OBJECT_NULL) {
        NSLog(@"Failed to retrieve GPU iterator from IOKit");
        [metalDevices release];
        return;
    }

    io_service_t ioDevice;
    while ((ioDevice = IOIteratorNext(gpuIterator)) != IO_OBJECT_NULL) {
        uint64_t ioRegistryID = GetRegistryIDForIOPCIDevice(ioDevice);
        NSString *ioDeviceName = (NSString *)IORegistryEntryCreateCFProperty(
            ioDevice, CFSTR(kIOProductNameKey), kCFAllocatorDefault, 0);
        
        // Check each Metal device for a match
        for (id<MTLDevice> metalDevice in metalDevices) {
            BOOL isMatch = NO;
            
            if (@available(macOS 10.13, *)) {
                // Use registryID for Metal 2+ (most modern systems)
                isMatch = (metalDevice.registryID == ioRegistryID);
            } else {
                // Fallback: match by device name for Metal 1/older Mac Pros
                isMatch = [ioDeviceName isEqualToString:metalDevice.name];
            }
            
            if (isMatch) {
                NSLog(@"Matched GPU:");
                NSLog(@"  Metal Device: %@ (ID: %llu)", metalDevice.name, metalDevice.registryID);
                NSLog(@"  IOKit Device: %@ (Registry ID: %llu)", ioDeviceName, ioRegistryID);
                // Here you can mark this IOPCIDevice as Metal-supported
                break;
            }
        }

        // Clean up IOKit/CF objects
        CFRelease(ioDeviceName);
        IOObjectRelease(ioDevice);
    }

    IOObjectRelease(gpuIterator);
    [metalDevices release];
}

Key Notes for Older Mac Pros #

• Metal 1 Fallback: Pre-macOS 10.13 systems (or Metal 1 devices) don't have registryID, so matching by device name is the most reliable alternative. For even more precision, you could compare PCI bus/device/function numbers from IOKit against Metal's PCIe properties (if available).
• Multiple GPUs: Older Mac Pros often have dual GPUs (integrated + discrete), so the loop will correctly match each one if it supports Metal.
• Memory Management: Always remember to release IOKit objects (IOObjectRelease) and CF/NS objects to avoid leaks—critical for long-running apps.

内容的提问来源于stack exchange，提问作者Mike97