diff --git a/python/paddle/fluid/dygraph/layers.py b/python/paddle/fluid/dygraph/layers.py index 3275a2126eddee8044ef156320f4b829fb209695..a9237e108049360dfe4a8f5fd7f80b103f9bf92a 100644 --- 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())