intel_unified_runtime_adapter_platform_device_discovery.py

1import intel_cmplr_lib_ur as ur
2
3def main():
4    # Initialize the Unified Runtime
5    # This loads available adapters (e.g., Level Zero, CUDA, OpenCL)
6    adapters = ur.get_adapters()
7    
8    if not adapters:
9        print("No Unified Runtime adapters found.")
10        return
11
12    print(f"Found {len(adapters)} adapter(s):")
13
14    for i, adapter in enumerate(adapters):
15        # Retrieve adapter information
16        backend = adapter.get_backend()
17        print(f"\nAdapter {i}: Backend - {backend}")
18
19        # Get platforms associated with this adapter
20        platforms = adapter.get_platforms()
21        print(f"  Found {len(platforms)} platform(s):")
22
23        for j, platform in enumerate(platforms):
24            name = platform.get_info(ur.platform_info_t.NAME)
25            vendor = platform.get_info(ur.platform_info_t.VENDOR)
26            print(f"    Platform {j}: {name} ({vendor})")
27
28            # Get devices for this platform
29            devices = platform.get_devices()
30            print(f"      Devices ({len(devices)}):")
31            for device in devices:
32                dev_name = device.get_info(ur.device_info_t.NAME)
33                dev_type = device.get_info(ur.device_info_t.TYPE)
34                print(f"        - {dev_name} [Type: {dev_type}]")
35
36if __name__ == "__main__":
37    main()