general/occlum
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix PyTorch demo CI hang

Browse Source
This commit is contained in:
Hui, Chunyang 2021-06-25 05:04:37 +00:00 committed by Zongmin.Gu
parent 09b7828f44
commit ea7b7b9fcf
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
.github/workflows/demo_test.yml vendored
View File

@ -314,6 +314,7 @@ jobs:
- name: Run Tensorflow-lite benchmark - name: Run Tensorflow-lite benchmark
run: docker exec tflite_test bash -c "cd /root/occlum/demos/tensorflow_lite && SGX_MODE=SIM ./run_tflite_in_occlum.sh benchmark" run: docker exec tflite_test bash -c "cd /root/occlum/demos/tensorflow_lite && SGX_MODE=SIM ./run_tflite_in_occlum.sh benchmark"
Pytorch_test: Pytorch_test:
runs-on: ubuntu-18.04 runs-on: ubuntu-18.04
steps: steps:
@ -337,7 +338,19 @@ jobs:
run: docker exec pytorch_test bash -c "cd /root/occlum/demos/pytorch; ./install_python_with_conda.sh" run: docker exec pytorch_test bash -c "cd /root/occlum/demos/pytorch; ./install_python_with_conda.sh"
- name: Run pytorch test - name: Run pytorch test
run: docker exec pytorch_test bash -c "cd /root/occlum/demos/pytorch; SGX_MODE=SIM ./run_pytorch_on_occlum.sh" run: docker exec -d pytorch_test bash -c "cd /root/occlum/demos/pytorch; SGX_MODE=SIM ./run_pytorch_on_occlum.sh 2>&1 | tee /root/occlum/log"
# FIXME: PyTorch can't exit normally in SIM mode
- name: Kill the container
run: |
sleep 180;
cat "$GITHUB_WORKSPACE/log";
if grep -q Done "$GITHUB_WORKSPACE/log"; then
docker stop pytorch_test
else
exit 1
fi
# Below tests needs test image to run faster # Below tests needs test image to run faster
Grpc_test: Grpc_test: