fix the state_dict bug for the xpu (#30008)

9859afa9 · wawltor · GitHub · faf2bb39 · 9859afa9
显示空白变更内容
内联并排

Showing with 4 addition and 0 deletion

python/paddle/fluid/dygraph/layers.py python/paddle/fluid/dygraph/layers.py +4 -0

未找到文件。
--- a/python/paddle/fluid/dygraph/layers.py
+++ b/python/paddle/fluid/dygraph/layers.py
@@ -1262,6 +1262,10 @@ class Layer(core.Layer):
                    place = core.CPUPlace()
                elif p.is_cuda_pinned_place():
                    place = core.CUDAPinnedPlace()
+                elif p.is_xpu_place():
+                    p = core.Place()
+                    p.set_place(t._place())
+                    place = core.XPUPlace(p.xpu_device_id())
                else:
                    p = core.Place()
                    p.set_place(t._place())