add torchtitan

Author: lowdy1

.github/workflows/_ascend_npu_torchtitan.yml

@@ -119,7 +119,7 @@ jobs:
       - name: Install project dependencies
         run: |
           pip install -r requirements.txt
-          pip install pytest pytest-cov
+          pip install pytest pytest-cov tyro
 
       - name: Show environment info
         run: |

.github/workflows/ascend_npu_test.yml

@@ -183,7 +183,7 @@ jobs:
       (success() || (needs.build-torch.result == 'skipped' && needs.build.result == 'success'))
     uses: ./.github/workflows/_ascend_npu_torchtitan.yml
     with:
-      runner: "linux-arm64-npu-2"
+      runner: "linux-arm64-npu-4"
       image: "ascendai/cann:8.0.0.beta1-910b-ubuntu22.04-py3.10"
       device:  ${{ needs.prepare.outputs.device }}
       torch-artifact: ${{ needs.build-torch.outputs.torch-artifact }}