deepspeech2 modify for kunlun

paddlespeech/s2t/exps/deepspeech2/bin/train.py

@@ -33,6 +33,11 @@ if __name__ == "__main__":
     parser = default_argument_parser()
     parser.add_argument(
         "--model_type", type=str, default='offline', help='offline/online')
+    parser.add_argument(
+        '--nxpu',
+        type=int,
+        default=1,
+        help="if nxpu == 0 and ngpu == 0, use cpu.")
     args = parser.parse_args()
     print("model_type:{}".format(args.model_type))
     print_arguments(args, globals())

paddlespeech/s2t/training/trainer.py

@@ -112,7 +112,14 @@ class Trainer():
         logger.info(f"Rank: {self.rank}/{self.world_size}")
 
         # set device
-        paddle.set_device('gpu' if self.args.ngpu > 0 else 'cpu')
+        if self.args.ngpu == 0:
+            paddle.set_device('cpu')
+            if self.args.nxpu == 0:
+                paddle.set_device('cpu')
+            else:
+                paddle.set_device('xpu')
+        elif self.args.ngpu > 0:
+            paddle.set_device("gpu")
         if self.parallel:
             self.init_parallel()