dbartlett/borg_velocity
0
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add HMC and BlackJax parameter samplers

Browse source
This commit is contained in:
Deaglan Bartlett 2024-10-25 11:40:58 +02:00
parent fca3d19059
commit 8a0c638e5b
29 changed files with 829 additions and 160 deletions
Download patch file Download diff file Expand all files Collapse all files

12
scripts/build_borg.sh
View file

@ -9,7 +9,10 @@ module load cuda/12.3
source /home/bartlett/.bashrc
source /home/bartlett/anaconda3/etc/profile.d/conda.sh
conda deactivate
conda activate borg_env
#conda activate borg_env
#conda activate borg
#conda activate compiler_test
conda activate borg_new
# Kill job if there are any errors
set -e
@ -21,9 +24,10 @@ BUILD_DIR=/data101/bartlett/build_borg/
cd $BORG_DIR
# git pull
# rm -rf $BUILD_DIR
bash build.sh --c-compiler $(which x86_64-conda_cos6-linux-gnu-gcc) --cxx-compiler $(which x86_64-conda_cos6-linux-gnu-g++) --python --hades-python --build-dir $BUILD_DIR
#bash build.sh --c-compiler $(which x86_64-conda_cos6-linux-gnu-gcc) --cxx-compiler $(which x86_64-conda_cos6-linux-gnu-g++) --python --hades-python --build-dir $BUILD_DIR
bash build.sh --c-compiler $(which x86_64-conda_cos6-linux-gnu-gcc) --cxx-compiler $(which x86_64-conda_cos6-linux-gnu-g++) --python=$(which python3) --install-system-python --hades-python --use-system-hdf5 --build-dir /data101/bartlett/build_borg/
cd $BUILD_DIR
make -j
#make -j 32
exit 0
exit 0

8
scripts/run_borg.sh
View file

@ -3,20 +3,22 @@
# Modules
module purge
module restore myborg
module load cuda/12.3
module load cuda/12.6
# Environment
source /home/bartlett/.bashrc
source /home/bartlett/anaconda3/etc/profile.d/conda.sh
conda deactivate
conda activate borg_env
#conda activate borg_env
#conda activate compiler_test
conda activate borg_new
# Kill job if there are any errors
set -e
# Path variables
BORG=/data101/bartlett/build_borg/tools/hades_python/hades_python
RUN_DIR=/data101/bartlett/fsigma8/borg_velocity/supranta_pars_N64
RUN_DIR=/data101/bartlett/fsigma8/borg_velocity/test_dir_blackjax
mkdir -p $RUN_DIR
cd $RUN_DIR

6
scripts/submit_borg.sh
View file

@ -1,9 +1,9 @@
#!/bin/sh
#PBS -S /bin/sh
#PBS -N supranta_pars_N64
#PBS -N supranta_N64_hmc_modelpar
#PBS -j oe
#PBS -m ae
#PBS -l nodes=h11:has1gpu:ppn=40,walltime=72:00:00
#PBS -l nodes=h10:has1gpu:ppn=40,walltime=5:00:00
# Modules
module purge
@ -21,7 +21,7 @@ set -e
# Path variables
BORG=/data101/bartlett/build_borg/tools/hades_python/hades_python
RUN_DIR=/data101/bartlett/fsigma8/borg_velocity/supranta_pars_N64
RUN_DIR=/data101/bartlett/fsigma8/borg_velocity/test_dir_hmc_modelpar
mkdir -p $RUN_DIR
cd $RUN_DIR