add npu device support for uie (#4401)

* add npu support for uie

* update get_device/get_env_device

Author: YanhuiDua

paddlenlp/taskflow/task.py

@@ -171,6 +171,9 @@ def _prepare_static_mode(self):
         if paddle.get_device() == "cpu":
             self._config.disable_gpu()
             self._config.enable_mkldnn()
+        elif paddle.get_device().split(":", 1)[0] == "npu":
+            self._config.disable_gpu()
+            self._config.enable_npu(self.kwargs["device_id"])
         else:
             self._config.enable_use_gpu(100, self.kwargs["device_id"])
             # TODO(linjieccc): enable after fixed

paddlenlp/utils/tools.py

@@ -122,7 +122,7 @@ def get_env_device():
     """
     if paddle.is_compiled_with_cuda():
         return "gpu"
-    elif paddle.is_compiled_with_npu():
+    elif "npu" in paddle.device.get_all_custom_device_type():
         return "npu"
     elif paddle.is_compiled_with_rocm():
         return "rocm"