intel_ipex_xpu_device_detection_and_tensor_verification.py

1import torch
2import intel_extension_for_pytorch as ipex
3
4def check_unified_runtime():
5    print(f"PyTorch Version: {torch.__version__}")
6    print(f"IPEX Version: {ipex.__version__}")
7
8    # Check if the Unified Runtime is correctly detecting hardware
9    if torch.xpu.is_available():
10        print("Success: Unified Runtime (UR) is active.")
11        print(f"Device Name: {torch.xpu.get_device_name(0)}")
12        
13        # Perform a simple tensor operation to verify the runtime backend
14        x = torch.ones((3, 3), device="xpu")
15        y = x + x
16        print("Tensor operation on XPU successful:")
17        print(y.cpu())
18    else:
19        print("Error: XPU device not found. Check intel-cmplr-lib-ur installation and drivers.")
20
21if __name__ == "__main__":
22    check_unified_runtime()