update

Author: shink

.github/workflows/_ascend_npu_build_torch_npu.yml

@@ -11,6 +11,10 @@ on:
         required: true
         type: string
         description: 'The docker image which will be used to build'
+      device:
+        required: true
+        type: string
+        description: 'The device selected to run on'
       torch-artifact:
         required: false
         type: string
@@ -33,8 +37,18 @@ jobs:
     runs-on: ${{ inputs.runner }}
     container:
       image: ${{ inputs.image }}
+      volumes:
+        - /usr/local/dcmi:/usr/local/dcmi
+        - /usr/local/bin/npu-smi:/usr/local/bin/npu-smi
+        - /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/
+        - /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info
+        - /etc/ascend_install.info:/etc/ascend_install.info
       options: >-
         --network host
+        --device ${{ inputs.device }}
+        --device /dev/davinci_manager
+        --device /dev/devmm_svm
+        --device /dev/hisi_hdc
       env:
         HTTP_PROXY: http://127.0.0.1:10809
         HTTPS_PROXY: http://127.0.0.1:10809