Add XPU to profiler

2026-09-19 17:24:32 +02:00 · 2025-03-11 17:43:48 +03:00
parent 30afbd036d
commit 9b7bb5f213
1 changed files with 2 additions and 0 deletions
@@ -188,6 +188,8 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
                activities=[torch.profiler.ProfilerActivity.CPU]
                if torch.cuda.is_available():
                    activities.append(torch.profiler.ProfilerActivity.CUDA)
+                if devices.has_xpu() and hasattr(torch.profiler.ProfilerActivity, "XPU"):
+                    activities.append(torch.profiler.ProfilerActivity.XPU)
                shared.log.debug(f'Torch profile: activities={activities}')
                if shared.profiler is None:
                    profile_args = {