test_compiler supports XPU.

Author: Xreki

graph_net/paddle/backend/cinn_backend.py

@@ -19,5 +19,6 @@ def synchronize(self):
         if (
             paddle.device.is_compiled_with_cuda()
             or paddle.device.is_compiled_with_rocm()
+            or paddle.device.is_compiled_with_xpu()
         ):
-            paddle.device.synchronize()
+            paddle.device.synchronize()

graph_net/paddle/backend/nope_backend.py

@@ -10,5 +10,6 @@ def synchronize(self):
         if (
             paddle.device.is_compiled_with_cuda()
             or paddle.device.is_compiled_with_rocm()
+            or paddle.device.is_compiled_with_xpu()
         ):
-            paddle.device.synchronize()
+            paddle.device.synchronize()

graph_net/paddle/test_compiler.py

@@ -12,6 +12,8 @@
 import random
 import platform
 import traceback
+import subprocess
+import re
 
 from graph_net.paddle import utils
 from graph_net import path_utils
@@ -40,12 +42,21 @@ def set_seed(random_seed):
 
 
 def get_hardward_name(args):
+    hardware = "unknown"
     if test_compiler_util.is_gpu_device(args.device):
         hardware = paddle.device.cuda.get_device_name(0)
+    elif args.device == "xpu":
+        try:
+            output = subprocess.check_output(["xpu-smi", "-L"], text=True)
+            hardware = next(
+                match.group(2)
+                for line in output.splitlines()
+                if (match := re.match(r"XPU\s+(\d+):\s+(.+?)\s+\(UUID:\s*([^)]+)\)", line))
+            )
+        except Exception as e:
+            pass
     elif args.device == "cpu":
         hardware = platform.processor()
-    else:
-        hardware = "unknown"
     return hardware
 
 
@@ -422,7 +433,7 @@ def test_multi_models(args):
 def main(args):
     assert os.path.isdir(args.model_path)
     assert args.compiler in {"cinn", "nope"}
-    assert args.device in ["cuda", "dcu", "cpu"]
+    assert args.device in ["cuda", "dcu", "xpu", "cpu"]
 
     initalize_seed = 123
     set_seed(random_seed=initalize_seed)

graph_net/paddle/utils.py

@@ -139,6 +139,7 @@ def convert_to_valid_number(data_type, value):
 
 
 def convert_meta_classes_to_tensors(file_path):
+    current_device = paddle.device.get_device()
     for name, cls in _get_classes(file_path):
         attrs = {
             k: v
@@ -159,7 +160,7 @@ def convert_meta_classes_to_tensors(file_path):
             "info": {
                 "shape": attrs.get("shape", []),
                 "dtype": data_type,
-                "device": attrs.get("device", "gpu"),
+                "device": attrs.get("device", current_device),
                 "mean": convert_to_valid_number(data_type, attrs.get("mean", None)),
                 "std": convert_to_valid_number(data_type, attrs.get("std", None)),
                 "min_val": convert_to_valid_number(data_type, attrs.get("min_val", 0)),