ECE 408 sem project

kernels_op_7/sweep.slurm

@@ -0,0 +1,29 @@
+#!/bin/bash
+#SBATCH --job-name="op7_sweep"
+#SBATCH --partition=gpuA40x4
+#SBATCH --mem=12G
+#SBATCH --nodes=1
+#SBATCH --ntasks-per-node=1
+#SBATCH --cpus-per-task=4
+#SBATCH --gpus-per-node=1
+#SBATCH --gpu-bind=closest
+#SBATCH --constraint="projects,perf,nvperf"
+#SBATCH --account=bche-delta-gpu
+#SBATCH --time=01:00:00
+#SBATCH --output=op7_sweep_%j.out
+#SBATCH --mail-user=asing271@illinois.edu
+#SBATCH --mail-type=BEGIN,END,FAIL
+
+module reset
+module load cuda
+
+cd "$SLURM_SUBMIT_DIR"
+
+# 1. Save current kernels and swap in req_4.
+BACKUP=$(mktemp -d)
+cp kernels/*.cuh "$BACKUP/"
+cp kernels_req_4/*.cuh kernels/
+
+# 2. Always restore kernels/ on exit.
+trap 'cp "$BACKUP"/*.cuh kernels/' EXIT
+python3 kernels_op_7/sweep.py