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

Before Change

    plaidml.settings._setup_for_test(plaidml.settings.user_settings)
    plaidml.settings.experimental = True
    devices, _ = plaidml.devices(ctx, limit=100, return_all=True)
    if gpu:
        for device in devices:
            if b"cuda" in device.description.lower():
                return device
        for device in devices:
            if b"opencl" in device.description.lower():
                return device
    for device in devices:
        if b"llvm" in device.description.lower():
            return device

After Change

            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