File size: 742 Bytes
2af0e94 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | #!/bin/bash -l
#SBATCH --job-name=xpu-verify-fix
#SBATCH --account=AIRR-P51-DAWN-GPU
#SBATCH --partition=pvc9
#SBATCH --nodes=1
#SBATCH --gres=gpu:1
#SBATCH --ntasks-per-node=1
#SBATCH --cpus-per-task=12
#SBATCH --time=01:00:00
#SBATCH --output=Logs/verify_fix_%j.out
#SBATCH --error=Logs/verify_fix_%j.err
. /etc/profile.d/modules.sh
module purge
module load rhel9/default-dawn
source ~/miniconda3/etc/profile.d/conda.sh
conda activate ~/rds/rds-airr-p51-TWhPgQVLKbA/Env/pub_env/pytorch-xpu
echo "=== Compare ConvTranspose3d (leaky) vs UpsampleConv (fix) vs full OM_net ==="
python tests/diagnose_xpu_leak_ops.py --test deconv
python tests/diagnose_xpu_leak_ops.py --test upsample_conv
python tests/diagnose_xpu_leak_ops.py --test om_net
|