Fix cluster startup script (#2977)

This PR fixes cluster startup script Authors: - Vibhu Jawa (https://github.com/VibhuJawa) Approvers: - Alex Barghi (https://github.com/alexbarghi-nv) - Brad Rees (https://github.com/BradReesWork) - Rick Ratzel (https://github.com/rlratzel) URL: #2977
rapidsai · Nov 30, 2022 · 60f5e7b · 60f5e7b
1 parent fb5298d
commit 60f5e7b
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
diff --git a/python/cugraph-service/scripts/default-config.sh b/python/cugraph-service/scripts/default-config.sh
@@ -21,6 +21,7 @@ THIS_DIR=$(cd $(dirname ${BASH_SOURCE[0]}) && pwd)
 # simply not use that syntax and override, since these variables are
 # read last.
 SCRIPTS_DIR=$THIS_DIR
+WORKSPACE=$THIS_DIR
 
 # These really should be oerridden by the project config!
 CONDA_ENV=${CONDA_ENV:-rapids}

diff --git a/python/cugraph-service/scripts/run-dask-process.sh b/python/cugraph-service/scripts/run-dask-process.sh
@@ -98,7 +98,6 @@ function buildTcpArgs {
     export DASK_DISTRIBUTED__WORKER__MEMORY__Terminate="False"
 
     SCHEDULER_ARGS="--protocol=tcp
-                    --port=$DASK_SCHEDULER_PORT
                     --scheduler-file $SCHEDULER_FILE
                 "
 
@@ -119,7 +118,6 @@ function buildUCXWithInfinibandArgs {
     export DASK_DISTRIBUTED__COMM__UCX__CREATE_CUDA_CONTEXT=True
 
     SCHEDULER_ARGS="--protocol=ucx
-                --port=$DASK_SCHEDULER_PORT
                 --interface=$DASK_CUDA_INTERFACE
                 --scheduler-file $SCHEDULER_FILE
                "
@@ -131,7 +129,7 @@ function buildUCXWithInfinibandArgs {
                 --scheduler-file=$SCHEDULER_FILE
                 --memory-limit=$DASK_HOST_MEMORY_LIMIT
                 --device-memory-limit=$DASK_DEVICE_MEMORY_LIMIT
-                --jit-unspill
+                --enable-jit-unspill
                 "
 }
 
@@ -152,7 +150,6 @@ function buildUCXwithoutInfinibandArgs {
 
 
     SCHEDULER_ARGS="--protocol=ucx
-            --port=$DASK_SCHEDULER_PORT
             --scheduler-file $SCHEDULER_FILE
             "
 
@@ -166,7 +163,7 @@ function buildUCXwithoutInfinibandArgs {
                 --scheduler-file=$SCHEDULER_FILE
                 --memory-limit=$DASK_HOST_MEMORY_LIMIT
                 --device-memory-limit=$DASK_DEVICE_MEMORY_LIMIT
-                --jit-unspill
+                --enable-jit-unspill
                 "
 }
 

diff --git a/python/cugraph-service/tests/test_mg_e2e.py b/python/cugraph-service/tests/test_mg_e2e.py
@@ -42,7 +42,7 @@ def mg_server():
     from cugraph_service_client.exceptions import CugraphServiceError
 
     server_process = None
-    host = "localhost"
+    host = "0.0.0.0"
     port = 9090
     client = CugraphServiceClient(host, port)