fix

flexflow · Jul 11, 2023 · 3f4ad5f · 3f4ad5f
1 parent a7524c4
commit 3f4ad5f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/align/test_all_operators.sh b/tests/align/test_all_operators.sh
@@ -4,7 +4,7 @@ eval "$(conda shell.bash hook)"
 rm -rf align/out
 
 function generate_ff_tensor(){
-    ./python/flexflow_python tests/align/align_create_tensor_ff.py -ll:py 1 -ll:gpu 1 -ll:fsize 5000 -ll:zsize 4096 -b 16 -o "$1"
+    ./python/legion_python tests/align/align_create_tensor_ff.py -ll:py 1 -ll:gpu 1 -ll:fsize 5000 -ll:zsize 4096 -b 16 -o "$1"
 }
 
 function generate_torch_tensor(){

diff --git a/tests/multinode_helpers/mpi_wrapper2.sh b/tests/multinode_helpers/mpi_wrapper2.sh
@@ -12,7 +12,7 @@ if [ -z "$GPUS" ]; then echo "GPUS variable is not defined, aborting tests"; exi
 CUDA_VISIBLE_DEVICES=$(seq -s, $((OMPI_COMM_WORLD_RANK * GPUS ))  $(( OMPI_COMM_WORLD_RANK * GPUS +1 )) )
 export CUDA_VISIBLE_DEVICES
 
-EXE="$FF_HOME"/python/flexflow_python
+EXE="$FF_HOME"/python/legion_python
 
 $EXE "$@"