Skip to content

Instantly share code, notes, and snippets.

@vamatya
Created October 23, 2019 20:39
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save vamatya/6fc3e2df20a03ba89d3cd03512ba8b12 to your computer and use it in GitHub Desktop.
Save vamatya/6fc3e2df20a03ba89d3cd03512ba8b12 to your computer and use it in GitHub Desktop.
#!/bin/bash
### Begin BSUB Options
#BSUB -P GEN135
#BSUB -J sgd_out
#BSUB -W 1:59
#BSUB -nnodes 1
#BSUB -alloc_flags "smt1"
### End BSUB Options and begin shell commands
export folder_name="/gen135"
export output_dir="$MEMBERWORK$folder_name"
export executable="/autofs/nccsopen-svm1_home/vamatya/vector-function/src/bm_apps/build/StandaloneProto"
for bus in 10
#100 1000 10000 10000 100000
do
for threads in 16
#2 4 8 16 32
do
outputfile=${output_dir}/standalone_Nbus_${bus}_threads_${threads}
#touch ${outputfile}
inputfile=${output_dir}/configuration.cfg.numBus_${bus}
matrixfile=${output_dir}/spmat.iajaa.numBus_${bus}
export OMP_NUM_THREADS=$threads
jsrun --nrs 1 --tasks_per_rs 1 --cpu_per_rs $threads --gpu_per_rs 1 --rs_per_host 1 --latency_priority CPU-CPU --launch_distribution packed --bind rs js_task_info $executable -f ${inputfile} -m ${matrixfile} > ${outputfile}
done
done
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment