fix the state_dict bug for the xpu (#29888)

fix the state_dict bug for the xpu

fix the state_dict bug for the xpu (#29888)
fix the state_dict bug for the xpu
587b67ef · wawltor · GitHub · 121658d2 · 587b67ef
隐藏空白更改
内联并排

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
@@ -1274,6 +1274,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())