Update deepspeed.yaml

Ascend · Dec 24, 2024 · ceb6b94 · ceb6b94
1 parent 4477b0b
commit ceb6b94
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/deepspeed.yaml b/.github/workflows/deepspeed.yaml
@@ -35,6 +35,7 @@ jobs:
         - /etc/ascend_install.info:/etc/ascend_install.info   
       options: --network host
                --name deepspeed_unit-tests
+               --device /dev/davinci0
                --device /dev/davinci4
                --device /dev/davinci_manager
                --device /dev/devmm_svm