Skip to content

[Manual] Devito on ARM. Tested on Raspberry Pi 4 5

George Bisbas edited this page Feb 3, 2025 · 1 revision

Playing on RPi4 Cortex-A72, 8 GB of memory Playing on RPi5 Cortex-A76, 8 GB of memory

export DEVITO_JIT_BACKDOOR=0
export DEVITO_LANGUGE=openmp
export DEVITO_LOGGING=DEBUG
export DEVITO_ARCH=arm
export DEVITO_PLATFORM=cortexa76
export DEVITO_COMPILER=arm
python examples/seismic/acoustic/acoustic_example.py 

Ask for an interactive job:

sbatch --nodes=1 --ntasks=1 --cpus-per-task=4 --time=00:01:00 --wrap="srun --pty bash"
source python311-venv/bin/activate

Benchmarks:

export OMP_NUM_THREADS=4
export DEVITO_ARCH=gcc
export DEVITO_LANGUAGE=openmp
export DEVITO_LOGGING=DEBUG
export DEVITO_PROFILING=advanced2
export DEVITO_MPI=1


# Biggest benchmarks possible:
export DEVITO_SAFE_HALO=1
python examples/seismic/acoustic/acoustic_example.py -d 540 540 540 --tn 128 -so 4 -a aggressive
python examples/seismic/acoustic/acoustic_example.py -d 540 540 540 --tn 128 -so 8 -a aggressive
python examples/seismic/acoustic/acoustic_example.py -d 540 540 540 --tn 128 -so 12 -a aggressive

export DEVITO_SAFE_HALO=2
python examples/seismic/elastic/elastic_example.py -d 300 300 300 --tn 128 -so 4 -a aggressive
python examples/seismic/elastic/elastic_example.py -d 300 300 300 --tn 128 -so 8 -a aggressive
python examples/seismic/elastic/elastic_example.py -d 300 300 300 --tn 128 -so 12 -a aggressive

export DEVITO_SAFE_HALO=1
python examples/seismic/tti/tti_example.py -d 340 340 340 --tn 128 -so 4 -a aggressive
python examples/seismic/tti/tti_example.py -d 340 340 340 --tn 128 -so 8 -a aggressive
python examples/seismic/tti/tti_example.py -d 340 340 340 --tn 128 -so 12 -a aggressive

export DEVITO_SAFE_HALO=2
python examples/seismic/viscoelastic/viscoelastic_example.py -d 240 240 240 --tn 128 -so 4 -a aggressive
python examples/seismic/viscoelastic/viscoelastic_example.py -d 240 240 240 --tn 128 -so 8 -a aggressive
python examples/seismic/viscoelastic/viscoelastic_example.py -d 240 240 240 --tn 128 -so 12 -a aggressive
Clone this wiki locally