From b0e5fd353c884d443aacffa18328971db9dd728a Mon Sep 17 00:00:00 2001
From: Fanli Lin <fanli.lin@intel.com>
Date: Thu, 31 Oct 2024 22:50:51 +0800
Subject: [PATCH] add xpu (#3163)

---
 src/accelerate/local_sgd.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/accelerate/local_sgd.py b/src/accelerate/local_sgd.py
index 382c34bc91c..47b8ed7c450 100644
--- a/src/accelerate/local_sgd.py
+++ b/src/accelerate/local_sgd.py
@@ -69,6 +69,7 @@ def __init__(self, accelerator: Accelerator, model: torch.nn.Module, local_sgd_s
             DistributedType.NO,
             DistributedType.MULTI_CPU,
             DistributedType.MULTI_GPU,
+            DistributedType.MULTI_XPU,
             DistributedType.MULTI_MLU,
             DistributedType.MULTI_MUSA,
             DistributedType.MULTI_NPU,