From 9b7bb5f213d70ee73f2135b64c07b69e373bd828 Mon Sep 17 00:00:00 2001
From: Disty0 <disty@disty.xyz>
Date: Tue, 11 Mar 2025 17:43:48 +0300
Subject: [PATCH] Add XPU to profiler

---
 modules/processing.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/modules/processing.py b/modules/processing.py
index 02eee0206..3a49f99ba 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -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 = {