update testpipeline

Author: staoxiao

src/diffusers/utils/testing_utils.py

@@ -1077,28 +1077,28 @@ def _is_torch_fp64_available(device):
     # Function definitions
     BACKEND_EMPTY_CACHE = {
         "cuda": torch.cuda.empty_cache,
-        # "xpu": torch.xpu.empty_cache,
+        "xpu": torch.xpu.empty_cache,
         "cpu": None,
         "mps": torch.mps.empty_cache,
         "default": None,
     }
     BACKEND_DEVICE_COUNT = {
         "cuda": torch.cuda.device_count,
-        # "xpu": torch.xpu.device_count,
+        "xpu": torch.xpu.device_count,
         "cpu": lambda: 0,
         "mps": lambda: 0,
         "default": 0,
     }
     BACKEND_MANUAL_SEED = {
         "cuda": torch.cuda.manual_seed,
-        # "xpu": torch.xpu.manual_seed,
+        "xpu": torch.xpu.manual_seed,
         "cpu": torch.manual_seed,
         "mps": torch.mps.manual_seed,
         "default": torch.manual_seed,
     }
     BACKEND_RESET_PEAK_MEMORY_STATS = {
         "cuda": torch.cuda.reset_peak_memory_stats,
-        # "xpu": getattr(torch.xpu, "reset_peak_memory_stats", None),
+        "xpu": getattr(torch.xpu, "reset_peak_memory_stats", None),
         "cpu": None,
         "mps": None,
         "default": None,
@@ -1112,7 +1112,7 @@ def _is_torch_fp64_available(device):
     }
     BACKEND_MAX_MEMORY_ALLOCATED = {
         "cuda": torch.cuda.max_memory_allocated,
-        # "xpu": getattr(torch.xpu, "max_memory_allocated", None),
+        "xpu": getattr(torch.xpu, "max_memory_allocated", None),
         "cpu": 0,
         "mps": 0,
         "default": 0,