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

Increase the pytorch demo TCS number to 64

Browse Source 
Also increase ready check delay time to 360s

Signed-off-by: Zheng, Qi <huaiqing.zq@antgroup.com>
This commit is contained in:
Zheng, Qi 2021-10-21 19:53:38 +08:00 committed by Zongmin.Gu
parent 8621e09dbb
commit e2d2cb375c
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -346,7 +346,7 @@ jobs:
# FIXME: PyTorch can't exit normally in SIM mode # FIXME: PyTorch can't exit normally in SIM mode
- name: Kill the container - name: Kill the container
run: | run: |
sleep 240; sleep 360;
cat "$GITHUB_WORKSPACE/log"; cat "$GITHUB_WORKSPACE/log";
if grep -q Done "$GITHUB_WORKSPACE/log"; then if grep -q Done "$GITHUB_WORKSPACE/log"; then
docker stop pytorch_test docker stop pytorch_test

2
demos/pytorch/run_pytorch_on_occlum.sh
View File

@ -17,7 +17,7 @@ fi
new_json="$(jq '.resource_limits.user_space_size = "6000MB" | new_json="$(jq '.resource_limits.user_space_size = "6000MB" |
.resource_limits.kernel_space_heap_size = "256MB" | .resource_limits.kernel_space_heap_size = "256MB" |
.process.default_mmap_size = "4000MB" | .resource_limits.max_num_of_threads = 64 |
.env.default += ["PYTHONHOME=/opt/python-occlum"]' Occlum.json)" && \ .env.default += ["PYTHONHOME=/opt/python-occlum"]' Occlum.json)" && \
echo "${new_json}" > Occlum.json echo "${new_json}" > Occlum.json
occlum build occlum build