Skip to content

Commit

Permalink
Update script to compile WarpX on Frontier (OLCF) (#5076)
Browse files Browse the repository at this point in the history
* update profile script for Frontier

* satisfy dependency for module load

* change module versions and manually install adios2

* fix bug

* fix bug

* update instructions

* update scripts

* remove line for LibEnsemble
  • Loading branch information
lucafedeli88 authored Jul 25, 2024
1 parent 49df0ca commit eda6af5
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 10 deletions.
14 changes: 7 additions & 7 deletions Tools/machines/frontier-olcf/frontier_warpx.profile.example
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,9 @@ if [ -z ${proj-} ]; then echo "WARNING: The 'proj' variable is not yet set in yo
# required dependencies
module load cmake/3.23.2
module load craype-accel-amd-gfx90a
module load rocm/5.2.0 # waiting for 5.6 for next bump
module load cray-mpich
module load cce/15.0.0 # must be loaded after rocm
module load rocm/5.7.1
module load cray-mpich/8.1.28
module load cce/17.0.0 # must be loaded after rocm

# optional: faster builds
module load ccache
Expand All @@ -26,14 +26,14 @@ export LD_LIBRARY_PATH=${HOME}/sw/frontier/gpu/blaspp-2024.05.31/lib64:$LD_LIBRA
export LD_LIBRARY_PATH=${HOME}/sw/frontier/gpu/lapackpp-2024.05.31/lib64:$LD_LIBRARY_PATH

# optional: for QED lookup table generation support
module load boost/1.79.0-cxx17
module load boost/1.79.0

# optional: for openPMD support
module load adios2/2.8.3
module load hdf5/1.14.0
module load adios2/2.8.3-mpi
module load hdf5/1.12.1-mpi

# optional: for Python bindings or libEnsemble
module load cray-python/3.9.13.1
module load cray-python/3.11.5

if [ -d "${HOME}/sw/frontier/gpu/venvs/warpx-frontier" ]
then
Expand Down
3 changes: 0 additions & 3 deletions Tools/machines/frontier-olcf/install_dependencies.sh
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,6 @@ CXX=$(which CC) CXXFLAGS="-DLAPACK_FORTRAN_ADD_" cmake -S $HOME/src/lapackpp -B
cmake --build $HOME/src/lapackpp-frontier-gpu-build --target install --parallel 16
rm -rf $HOME/src/lapackpp-frontier-gpu-build


# Python ######################################################################
#
python3 -m pip install --upgrade pip
Expand Down Expand Up @@ -109,8 +108,6 @@ CUPY_INSTALL_USE_HIP=1 \
ROCM_HOME=${ROCM_PATH} \
HCC_AMDGPU_TARGET=${AMREX_AMD_ARCH} \
python3 -m pip install -v cupy
# optional: for libEnsemble
python3 -m pip install -r $HOME/src/warpx/Tools/LibEnsemble/requirements.txt
# optional: for optimas (based on libEnsemble & ax->botorch->gpytorch->pytorch)
#python3 -m pip install --upgrade torch --index-url https://download.pytorch.org/whl/rocm5.4.2
#python3 -m pip install -r $HOME/src/warpx/Tools/optimas/requirements.txt

0 comments on commit eda6af5

Please sign in to comment.