fix the state_dict bug for the xpu (#29888)

fix the state_dict bug for the xpu
4 years ago · 587b67ef62
parent 121658d251
commit 587b67ef62
1 changed files with 4 additions and 0 deletions
--- 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())