From 280d9d21c9081bfe237ee431d869107499856c34 Mon Sep 17 00:00:00 2001 From: Carl Pearson Date: Mon, 11 Dec 2023 10:08:40 -0700 Subject: [PATCH] some SPR hacks for oneMKL --- blake-spr/download-worker.sh | 2 +- blake-spr/env.sh | 3 ++- blake-spr/run.sh | 3 ++- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/blake-spr/download-worker.sh b/blake-spr/download-worker.sh index bbff4e7..853feda 100755 --- a/blake-spr/download-worker.sh +++ b/blake-spr/download-worker.sh @@ -12,7 +12,7 @@ mkdir -p "$LOG_DIR" # 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 +git clone "$KOKKOS_REMOTE" "$KERNELS_SRC" || true (cd "$KERNELS_SRC" && git checkout $KERNELS_SHA) || true # re-set up our environment diff --git a/blake-spr/env.sh b/blake-spr/env.sh index c3c7895..0e6a715 100644 --- a/blake-spr/env.sh +++ b/blake-spr/env.sh @@ -5,7 +5,8 @@ export KOKKOS_SRC="$ROOT_DIR/kokkos-${KOKKOS_SHA:0:8}" export KOKKOS_BUILD="$ROOT_DIR/kokkos-build-${KOKKOS_SHA:0:8}" export KOKKOS_INSTALL="$ROOT_DIR/kokkos-install-${KOKKOS_SHA:0:8}" -export KERNELS_SHA=a80eb9114ddda2d9454e4f3cc8a3dd5143ecdfc8 # 2023 11 30 +export KERNELS_REMOTE="git@github.com:cwpearson/kokkos-kernels.git" +export KERNELS_SHA=a91a1f24560692d6a294e764f82f497dc10aeb7d # 2023 12 08 export KERNELS_SRC="$ROOT_DIR/kernels-${KERNELS_SHA:0:8}" export KERNELS_BUILD="$ROOT_DIR/kernels-build-${KERNELS_SHA:0:8}" diff --git a/blake-spr/run.sh b/blake-spr/run.sh index 9737070..447b9dd 100755 --- a/blake-spr/run.sh +++ b/blake-spr/run.sh @@ -14,5 +14,6 @@ srun -n1 -t 1 hostname |& tee "$LOG_DIR/hostname.log" || true srun -n1 -t 1 cat /proc/cpuinfo |& tee "$LOG_DIR/cpuinfo.log" || true srun -n1 -t 1 env |& tee "$LOG_DIR/env.log" || true +srun -N 1 -p all -n1 -t 60 "$KERNELS_BUILD"/perf_test/sparse/KokkosKernels_sparse_spmv_benchmark -f /projects/cwpears/kug-2023/dielFilterV3real/dielFilterV3real.mtx |& tee "$LOG_DIR/spmv2.log" srun -N 1 -p all -n1 -t 60 "$KERNELS_BUILD"/perf_test/blas/blas3/KokkosKernels_Blas3_gemm_benchmark |& tee "$LOG_DIR/gemm.log" -srun -N 1 -p all -n1 -t 60 "$KERNELS_BUILD"/perf_test/sparse/KokkosKernels_sparse_spmv_benchmark -f /projects/cwpears/sparc_gpu_problems/single_gpu/matrix.mm |& tee "$LOG_DIR/spmv.log" \ No newline at end of file +srun -N 1 -p all -n1 -t 60 "$KERNELS_BUILD"/perf_test/sparse/KokkosKernels_sparse_spmv_benchmark -f /projects/cwpears/sparc_gpu_problems/single_gpu/matrix.mm |& tee "$LOG_DIR/spmv1.log"