From 3d0c0eb59db5a2399f8b78a2499ed8d056d547a4 Mon Sep 17 00:00:00 2001
From: baocheny <baochen.yang@intel.com>
Date: Tue, 27 Jun 2023 11:14:14 +0800
Subject: [PATCH] add 2 more custom devices intel_gpu and apple mps

---
 ppcls/static/train.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/ppcls/static/train.py b/ppcls/static/train.py
index 64e0e35d..898ebb02 100755
--- a/ppcls/static/train.py
+++ b/ppcls/static/train.py
@@ -90,8 +90,9 @@ def main(args):
         fleet.init(is_collective=True)
 
     # assign the device
-    assert global_config[
-        "device"] in ["cpu", "gpu", "xpu", "npu", "mlu", "ascend"]
+    assert global_config["device"] in [
+        "cpu", "gpu", "xpu", "npu", "mlu", "ascend", "intel_gpu", "mps"
+    ]
     device = paddle.set_device(global_config["device"])
 
     # amp related config
-- 
GitLab