guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-05-15 12:31:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

format

Browse source
This commit is contained in:
Wassim KABALAN 2024-10-22 12:57:34 -04:00
parent 86233081e2
commit 82b8f563a0
4 changed files with 60 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

4
benchmarks/particle_mesh.slurm
View file

@ -1,6 +1,6 @@
#!/bin/bash
##############################################################################################################################
# USAGE:sbatch --account=tkc@a100 --nodes=1 --gres=gpu:1 --tasks-per-node=1 -C a100 benchmarks/particle_mesh_a100.slurm
# USAGE:sbatch --account=tkc@a100 --nodes=1 --gres=gpu:1 --tasks-per-node=1 -C a100 benchmarks/particle_mesh_a100.slurm
##############################################################################################################################
#SBATCH --job-name=Particle-Mesh # nom du job
#SBATCH --cpus-per-task=8 # nombre de CPU par tache pour gpu_p5 (1/8 du noeud 8-GPU)
@ -140,7 +140,7 @@ fi
# GPU name is a100 if num_gpu_per_node is 8, otherwise it is v100
out_dir="pm_prof/$gpu_name/$nb_gpus"
trace_dir="traces/$gpu_name/$nb_gpus/bench_pm"
echo "Output dir is : $out_dir"
echo "Output dir is : $out_dir"
echo "Trace dir is : $trace_dir"
for pr in "${precisions[@]}"; do

4
benchmarks/pmwd_pm.slurm
View file

@ -1,6 +1,6 @@
#!/bin/bash
##############################################################################################################################
# USAGE:sbatch --account=tkc@a100 --nodes=1 --gres=gpu:1 --tasks-per-node=1 -C a100 benchmarks/particle_mesh_a100.slurm
# USAGE:sbatch --account=tkc@a100 --nodes=1 --gres=gpu:1 --tasks-per-node=1 -C a100 benchmarks/particle_mesh_a100.slurm
##############################################################################################################################
#SBATCH --job-name=Particle-Mesh # nom du job
#SBATCH --cpus-per-task=8 # nombre de CPU par tache pour gpu_p5 (1/8 du noeud 8-GPU)
@ -126,7 +126,7 @@ fi
out_dir="pm_prof/$gpu_name/$nb_gpus"
trace_dir="traces/$gpu_name/$nb_gpus/bench_pmwd"
echo "Output dir is : $out_dir"
echo "Output dir is : $out_dir"
echo "Trace dir is : $trace_dir"
for pr in "${precisions[@]}"; do