Skip to content

Commit 2fcbff8

Browse files
committed
add torchtitan
1 parent 52e285a commit 2fcbff8

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

.github/workflows/_ascend_npu_torchtitan.yml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,11 +130,13 @@ jobs:
130130
pip list | grep -E 'torch|numpy'
131131
132132
- name: Run torchtitan integration_test
133+
working-directory: torchtitan
133134
run: |
134135
mkdir artifacts-to-be-uploaded
135136
npu_count=$(python -c "import torch; print(torch.npu.device_count())")
136137
python ./tests/integration_tests.py artifacts-to-be-uploaded --ngpu ${npu_count}
137138
138139
- name: Run torchtitan unittest
140+
working-directory: torchtitan
139141
run: |
140142
pytest ./tests/unit_tests --cov=. --cov-report=xml --durations=20 -vv

.github/workflows/ascend_npu_test.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ on:
5656
- ascendai/cann:7.0.1-910b-ubuntu22.04-py3.8
5757
- ascendai/cann:8.0.0-910b-ubuntu22.04-py3.10
5858
- ascendai/cann:latest
59-
default: "ascendai/cann:7.0.1-910b-ubuntu22.04-py3.8"
59+
default: "ascendai/cann:latest"
6060
description: "The docker image which will be loaded"
6161
device:
6262
required: true

0 commit comments

Comments
 (0)