From e2d2cb375c51ddc5a1952742b3bb0b6b62ad6c78 Mon Sep 17 00:00:00 2001 From: "Zheng, Qi" Date: Thu, 21 Oct 2021 19:53:38 +0800 Subject: [PATCH] Increase the pytorch demo TCS number to 64 Also increase ready check delay time to 360s Signed-off-by: Zheng, Qi --- .github/workflows/demo_test.yml | 2 +- demos/pytorch/run_pytorch_on_occlum.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/demo_test.yml b/.github/workflows/demo_test.yml index 1001b45e..20e35dd5 100644 --- a/.github/workflows/demo_test.yml +++ b/.github/workflows/demo_test.yml @@ -346,7 +346,7 @@ jobs: # FIXME: PyTorch can't exit normally in SIM mode - name: Kill the container run: | - sleep 240; + sleep 360; cat "$GITHUB_WORKSPACE/log"; if grep -q Done "$GITHUB_WORKSPACE/log"; then docker stop pytorch_test diff --git a/demos/pytorch/run_pytorch_on_occlum.sh b/demos/pytorch/run_pytorch_on_occlum.sh index f701209b..7d964704 100755 --- a/demos/pytorch/run_pytorch_on_occlum.sh +++ b/demos/pytorch/run_pytorch_on_occlum.sh @@ -17,7 +17,7 @@ fi new_json="$(jq '.resource_limits.user_space_size = "6000MB" | .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)" && \ echo "${new_json}" > Occlum.json occlum build