fix bug that FLAGS_selected_xpus can't work (#1811)

PaddlePaddle · Mar 21, 2022 · 08a2275 · 08a2275
1 parent 6ff5eab
commit 08a2275
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/machine_translation/transformer/static/train.py b/examples/machine_translation/transformer/static/train.py
@@ -99,6 +99,7 @@ def do_train(args):
             places = paddle.static.cuda_places()
         elif args.device == "xpu":
             places = paddle.static.xpu_places()
+            paddle.set_device("xpu")
         else:
             places = paddle.static.cpu_places()
             paddle.set_device("cpu")