check_torch_dependencies

hlky · hlky · commit c8661f0dee75 · 2025-01-07T15:21:30.000Z
diff --git a/src/diffusers/utils/testing_utils.py b/src/diffusers/utils/testing_utils.py
@@ -1085,7 +1085,7 @@ def _is_torch_fp64_available(device):
     }
     BACKEND_RESET_PEAK_MEMORY_STATS = {
         "cuda": torch.cuda.reset_peak_memory_stats,
-        "xpu": torch.xpu.reset_peak_memory_stats,
+        "xpu": getattr(torch.xpu, "reset_peak_memory_stats", None),
         "default": None,
     }
     BACKEND_RESET_MAX_MEMORY_ALLOCATED = {
@@ -1094,7 +1094,7 @@ def _is_torch_fp64_available(device):
     }
     BACKEND_MAX_MEMORY_ALLOCATED = {
         "cuda": torch.cuda.max_memory_allocated,
-        "xpu": torch.xpu.max_memory_allocated,
+        "xpu": getattr(torch.xpu, "max_memory_allocated", None),
         "default": 0,
     }
 

Original file line number	Diff line number	Diff line change
`@@ -1085,7 +1085,7 @@ def _is_torch_fp64_available(device):`
`1085`	`1085`	`}`
`1086`	`1086`	`BACKEND_RESET_PEAK_MEMORY_STATS = {`
`1087`	`1087`	`"cuda": torch.cuda.reset_peak_memory_stats,`
`1088`		`- "xpu": torch.xpu.reset_peak_memory_stats,`
	`1088`	`+ "xpu": getattr(torch.xpu, "reset_peak_memory_stats", None),`
`1089`	`1089`	`"default": None,`
`1090`	`1090`	`}`
`1091`	`1091`	`BACKEND_RESET_MAX_MEMORY_ALLOCATED = {`
`@@ -1094,7 +1094,7 @@ def _is_torch_fp64_available(device):`
`1094`	`1094`	`}`
`1095`	`1095`	`BACKEND_MAX_MEMORY_ALLOCATED = {`
`1096`	`1096`	`"cuda": torch.cuda.max_memory_allocated,`
`1097`		`- "xpu": torch.xpu.max_memory_allocated,`
	`1097`	`+ "xpu": getattr(torch.xpu, "max_memory_allocated", None),`
`1098`	`1098`	`"default": 0,`
`1099`	`1099`	`}`
`1100`	`1100`