#!/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