45 lines
1.3 KiB
Bash
Executable File
45 lines
1.3 KiB
Bash
Executable File
#! /bin/bash
|
|
|
|
set -eou pipefail
|
|
|
|
source "$( cd -- "$(dirname "$0")" >/dev/null 2>&1 ; pwd -P )"/env.sh
|
|
|
|
LOG_DIR="$(date +"%Y%m%d_%H%M%S")_build"
|
|
export LOG_DIR
|
|
mkdir -p "$LOG_DIR"
|
|
|
|
# intel blows up SSH for some reason?
|
|
module del intel/oneAPI/hpc-toolkit/2022.1.2
|
|
git clone git@github.com:kokkos/kokkos.git "$KOKKOS_SRC" || true
|
|
(cd "$KOKKOS_SRC" && git checkout $KOKKOS_SHA) || true
|
|
git clone git@github.com:kokkos/kokkos-kernels.git "$KERNELS_SRC" || true
|
|
(cd "$KERNELS_SRC" && git checkout $KERNELS_SHA) || true
|
|
|
|
# re-set up our environment
|
|
source "$( cd -- "$(dirname "$0")" >/dev/null 2>&1 ; pwd -P )"/env.sh
|
|
|
|
module list |& tee "$LOG_DIR/module-list.log"
|
|
lscpu |& tee "$LOG_DIR/lscpu.log"
|
|
hostname |& tee "$LOG_DIR/hostname.log"
|
|
|
|
## Configure Kokkos
|
|
cmake -S "$KOKKOS_SRC" -B "$KOKKOS_BUILD" \
|
|
-DCMAKE_INSTALL_PREFIX="$KOKKOS_INSTALL" \
|
|
-DCMAKE_CXX_STANDARD=17 \
|
|
-DCMAKE_BUILD_TYPE=Release \
|
|
-DCMAKE_CXX_COMPILER=icpx \
|
|
-DKokkos_ENABLE_OPENMP=ON \
|
|
-DKokkos_ARCH_NATIVE=ON
|
|
|
|
## Build & Install Kokkos
|
|
cmake --build "$KOKKOS_BUILD" -j "$(nproc)" -t install
|
|
|
|
## Configure Kernels
|
|
cmake -S "$KERNELS_SRC" -B "$KERNELS_BUILD" \
|
|
-DKokkos_DIR="$KOKKOS_INSTALL/lib64/cmake/Kokkos" \
|
|
-DCMAKE_BUILD_TYPE=Release \
|
|
-DCMAKE_CXX_COMPILER=icpx \
|
|
-DKokkosKernels_ENABLE_TPL_MKL=ON
|
|
|
|
## Build Kernels
|
|
cmake --build "$KERNELS_BUILD" -j "$(nproc)" |