30 lines
717 B
Bash
30 lines
717 B
Bash
#!/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
|