Fix GPU bench scripts

2024-02-22 15:16:35 +01:00
parent f4747bfc9d
commit fc31299d1e
2 changed files with 5 additions and 3 deletions
--- a/experiments/reduce_bench_hemera_gpu.sh
+++ b/experiments/reduce_bench_hemera_gpu.sh
@ -14,9 +14,10 @@ cd $HOME/repos/metagraph_optimization

 module load singularity
 module load git 
+module load cuda/12.1

 printf "Current git commit hash: " > results/git_reduce_bench_gpu.txt
 git rev-parse HEAD >> results/git_reduce_bench_gpu.txt
 git status >> results/git_reduce_bench_gpu.txt

-singularity exec experiments/CUDA_container.sif ./experiments/run_reduce_bench_gpu.sh
+singularity exec --nv experiments/CUDA_container.sif ./experiments/run_reduce_bench_gpu.sh