Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
47 changes: 47 additions & 0 deletions hpc/batch_gpu/submit_imaging_delaunay_a100_hst_fp64
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
#!/bin/bash -l
#
# A100 first-class search profiling: searches/nautilus/imaging/delaunay × hst × fp64.
#
# Drives af.Nautilus end-to-end on an MGE-lens + Hilbert-image-mesh +
# Delaunay-source pixelization model with ConstantSplit regularization.
# Adapt image is the truth-derived `lensed_source.fits` cached next to
# the dataset (see searches/README.md "Datacube" / "adapt image" notes).

#SBATCH -J search_nautilus_imaging_delaunay_hst_fp64
#SBATCH --partition=gpu
#SBATCH --gres=gpu:1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=4
#SBATCH --mem=64gb
#SBATCH --time=3:00:00
#SBATCH -o output/output.%A.out
#SBATCH -e error/error.%A.err
#SBATCH --mail-type=END,FAIL
#SBATCH --mail-user=james.w.nightingale@durham.ac.uk

export AP_ROOT=/mnt/ral/jnightin/autolens_profiling
source $AP_ROOT/activate.sh

export JAX_PLATFORM_NAME=cuda
export JAX_PLATFORMS=cuda,cpu
export XLA_PYTHON_CLIENT_PREALLOCATE=false
export JAX_ENABLE_X64=True
export NUMBA_CACHE_DIR=/tmp/numba_cache
export MPLCONFIGDIR=/tmp/matplotlib

nvidia-smi

echo "=========================================="
date
echo "Cell: searches/nautilus/imaging/delaunay"
echo "Instrument: hst"
echo "Precision: fp64"

cd $AP_ROOT
python3 searches/nautilus/imaging/delaunay.py \
--instrument hst \
--config-name hpc_a100_fp64 \
--output-dir $AP_ROOT/results/searches/nautilus/imaging/delaunay/hst

echo "Finished."
date
45 changes: 45 additions & 0 deletions hpc/batch_gpu/submit_nss_imaging_delaunay_a100_hst_fp64
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
#!/bin/bash -l
#
# A100 first-class search profiling: searches/nss/imaging/delaunay × hst × fp64.
#
# Pair to submit_imaging_delaunay_a100_hst_fp64 (Nautilus); same dataset /
# model / analysis so timing and convergence numbers compare directly.

#SBATCH -J search_nss_imaging_delaunay_hst_fp64
#SBATCH --partition=gpu
#SBATCH --gres=gpu:1
#SBATCH --ntasks=1
#SBATCH --cpus-per-task=4
#SBATCH --mem=64gb
#SBATCH --time=3:00:00
#SBATCH -o output/output.%A.out
#SBATCH -e error/error.%A.err
#SBATCH --mail-type=END,FAIL
#SBATCH --mail-user=james.w.nightingale@durham.ac.uk

export AP_ROOT=/mnt/ral/jnightin/autolens_profiling
source $AP_ROOT/activate.sh

export JAX_PLATFORM_NAME=cuda
export JAX_PLATFORMS=cuda,cpu
export XLA_PYTHON_CLIENT_PREALLOCATE=false
export JAX_ENABLE_X64=True
export NUMBA_CACHE_DIR=/tmp/numba_cache
export MPLCONFIGDIR=/tmp/matplotlib

nvidia-smi

echo "=========================================="
date
echo "Cell: searches/nss/imaging/delaunay"
echo "Instrument: hst"
echo "Precision: fp64"

cd $AP_ROOT
python3 searches/nss/imaging/delaunay.py \
--instrument hst \
--config-name hpc_a100_fp64 \
--output-dir $AP_ROOT/results/searches/nss/imaging/delaunay/hst

echo "Finished."
date
Loading