test.sh

#!/bin/bash
#SBATCH -N 8
#SBATCH -C cpu
#SBATCH -q debug 
#SBATCH -t 00:08:00

. env.sh
export NODE_COUNT=4
export OMP_NUM_THREADS=16
export OMP_PLACES=threads

export CORES_PER_NODE=128
export RANKS_PER_NODE=$((CORES_PER_NODE / OMP_NUM_THREADS))

srun -np 3 python decompose.py -i uber \
                    --trank 5 \
                    -s 65000 \
                    -iter 40 \
                    -alg sts_cp \
                    -dist accumulator_stationary \
                    -r 1 #\
                    #-o data/fit_progress_vs_time \

#srun -N 4 -n 32 -c 32 python decompose.py -i caida \
#                    --trank 25 \
#                    -iter 10 \
#                    -alg cp_arls_lev \
#                    -s 4000000 \
#                    -dist accumulator_stationary \
#                    -p exact

#TENSOR=reddit
#srun -N $NODE_COUNT -n $((NODE_COUNT * RANKS_PER_NODE)) -c $((OMP_NUM_THREADS * 2)) \
#                    python decompose.py -i $TENSOR \
#                    --trank 25 \
#                    -iter 10 \
#                    -alg sts_cp \
#                    -s 65536 \
#                    -dist accumulator_stationary #\
                    #-p exact