Browse Source

Make run-clientscale-experiments and run-corescale-experiments a little more robust to the number of cores available

Ian Goldberg 4 months ago
parent
commit
c83606d135
2 changed files with 18 additions and 8 deletions
  1. 9 3
      docker/run-clientscale-experiments
  2. 9 5
      docker/run-corescale-experiments

+ 9 - 3
docker/run-clientscale-experiments

@@ -14,9 +14,15 @@ else
 fi
 
 if [ "$2" == "" ]; then
-    ncores=1
+    num_send_rounds=1
 else
-    ncores="$2"
+    num_send_rounds="$2"
+fi
+
+# Use up to 16 cores
+ncores=`nproc`
+if [ $ncores -ge 16 ]; then
+    ncores=16
 fi
 
 for iter in $(seq 1 $niters); do
@@ -25,7 +31,7 @@ for iter in $(seq 1 $niters); do
         docker exec -it ${SPARTA_DOCKER_PREFIX}sparta \
             numactl -C 0-$((ncores-1)) ftxsgx-runner \
             sparta/target/x86_64-fortanix-unknown-sgx/release/sparta.sgxs \
-            -- $sends $sends $ncores $sends 5 -r 10 -w 0
+            -- $sends $sends $ncores $sends 5 -r $num_send_rounds -w 0
     done
 done
 

+ 9 - 5
docker/run-corescale-experiments

@@ -19,13 +19,17 @@ else
     sends="$2"
 fi
 
+maxcores=`nproc`
+
 for iter in $(seq 1 $niters); do
     for ncores in 4 6 8 16 24 32 36 40 44 48 64 72; do
-        echo "cores: $ncores"
-        docker exec -it ${SPARTA_DOCKER_PREFIX}sparta \
-            numactl -C 0-$((ncores-1)) ftxsgx-runner \
-            sparta/target/x86_64-fortanix-unknown-sgx/release/sparta.sgxs \
-            -- $sends $sends $ncores $sends 5 -r 1 -w 0
+        if [ $ncores -le $maxcores ]; then
+            echo "cores: $ncores"
+            docker exec -it ${SPARTA_DOCKER_PREFIX}sparta \
+                numactl -C 0-$((ncores-1)) ftxsgx-runner \
+                sparta/target/x86_64-fortanix-unknown-sgx/release/sparta.sgxs \
+                -- $sends $sends $ncores $sends 5 -r 1 -w 0
+        fi
     done
 done