From b021a7697bfb5b039353860ff78c592ec5b0453c Mon Sep 17 00:00:00 2001
From: YAO Matrix <matrix.yao@intel.com>
Date: Tue, 1 Apr 2025 20:14:28 -0700
Subject: [PATCH] map BACKEND_RESET_MAX_MEMORY_ALLOCATED to
 reset_peak_memory_stats on XPU

Signed-off-by: YAO Matrix <matrix.yao@intel.com>
---
 src/diffusers/utils/testing_utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/diffusers/utils/testing_utils.py b/src/diffusers/utils/testing_utils.py
index 137420945340..e62f245f9ed1 100644
--- a/src/diffusers/utils/testing_utils.py
+++ b/src/diffusers/utils/testing_utils.py
@@ -1161,7 +1161,7 @@ def _is_torch_fp64_available(device):
     }
     BACKEND_RESET_MAX_MEMORY_ALLOCATED = {
         "cuda": torch.cuda.reset_max_memory_allocated,
-        "xpu": None,
+        "xpu": getattr(torch.xpu, "reset_peak_memory_stats", None),
         "cpu": None,
         "mps": None,
         "default": None,