File tree Expand file tree Collapse file tree 2 files changed +3
-1
lines changed Expand file tree Collapse file tree 2 files changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -130,11 +130,13 @@ jobs:
130
130
pip list | grep -E 'torch|numpy'
131
131
132
132
- name : Run torchtitan integration_test
133
+ working-directory : torchtitan
133
134
run : |
134
135
mkdir artifacts-to-be-uploaded
135
136
npu_count=$(python -c "import torch; print(torch.npu.device_count())")
136
137
python ./tests/integration_tests.py artifacts-to-be-uploaded --ngpu ${npu_count}
137
138
138
139
- name : Run torchtitan unittest
140
+ working-directory : torchtitan
139
141
run : |
140
142
pytest ./tests/unit_tests --cov=. --cov-report=xml --durations=20 -vv
Original file line number Diff line number Diff line change 56
56
- ascendai/cann:7.0.1-910b-ubuntu22.04-py3.8
57
57
- ascendai/cann:8.0.0-910b-ubuntu22.04-py3.10
58
58
- ascendai/cann:latest
59
- default : " ascendai/cann:7.0.1-910b-ubuntu22.04-py3.8 "
59
+ default : " ascendai/cann:latest "
60
60
description : " The docker image which will be loaded"
61
61
device :
62
62
required : true
You can’t perform that action at this time.
0 commit comments