b331b9f29062a0e888dafc7a7d19ca8fa53286c1,invesalius/segmentation/brain/utils.py,,get_plaidml_devices,#,1

Before Change

                return device
    for device in devices:
        if b"llvm" in device.description.lower():
            return device

After Change

    plaidml.settings.experimental = True
    devices, _ = plaidml.devices(ctx, limit=100, return_all=True)
    out_devices = []
    for device in devices:
        points = 0
        if b"cuda" in device.description.lower():
            points += 1
        if b"opencl" in device.description.lower():
            points += 1
        if b"nvidia" in device.description.lower():
            points += 1
        if b"amd" in device.description.lower():
            points += 1
        out_devices.append((points, device))

    out_devices.sort(reverse=True)
    return {device.description.decode("utf8"): device.id.decode("utf8") for points, device in out_devices }

Instances