add psc jupyter

Author: Johannes Blaschke

activate.sh

@@ -1 +1,8 @@
-ml use /global/common/software/nersc9/julia/modules/
+if [[ ${HOSTNAME} =~ ^.*bridges2.psc.edu$ ]]
+then
+    echo "I'm on PSC Bridges2 => Activating pre-production modules"
+    ml use /opt/packages/julia/modules
+else
+    echo "I'm on NERSC Perlmutter => Activating pre-production modules"
+    ml use /global/common/software/nersc9/julia/modules 
+fi

install.sh

@@ -1,22 +1,62 @@
-#!/usr/bin/env bash
+#!/usr/bin/env sh
 set -eu
 
 __INSTALL_SH_DIR=$(
     cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd
 )
 
-source ${__INSTALL_SH_DIR}/activate.sh
-
-mkdir -p ~/.local/share/jupyter/kernels
-pushd ~/.local/share/jupyter/kernels
-    # cp -r /global/common/software/nersc9/julia/kernels/rendered/*-beta .
-    cp -r /global/common/software/nersc/julia_hpc_24/kernels/* .
-popd
+JUPYTER_KERNEL_DIR=${__INSTALL_SH_DIR}/psc
+declare -a JUPYTER_KERNELS=(
+    "julia-tutorial-single-threaded"
+    "julia-tutorial-multi-threaded"
+)
+declare -a JUPYTER_THREADS_CT=(
+    "1"
+    "16"
+)
+declare -a JUPYTER_THREADS_NAME=(
+    "Single"
+    "Multi"
+)
 
-julia_versions=("julia/1.8.5" "julia/1.9.4" "julia/1.10.8" "julia/1.11.3")
-for jv in "${julia_versions[@]}"
+for kernel in {0..1}
 do
-    ml load $jv
-    julia /global/common/software/nersc9/julia/kernels/bootstrap.jl
+    kernel_dir=${HOME}/.local/share/jupyter/kernels/${JUPYTER_KERNELS[$kernel]}
+    mkdir -p ${kernel_dir}
+    echo "Creating Kernel at: ${kernel_dir}"
+
+    THREADS_NAME=${JUPYTER_THREADS_NAME[$kernel]} \
+    ${__INSTALL_SH_DIR}/lib/mo \
+        ${__INSTALL_SH_DIR}/psc/jupyter/template/kernel.json \
+        > ${kernel_dir}/kernel.json
+
+    THREADS_CT=${JUPYTER_THREADS_CT[$kernel]} \
+    NERSC_RESOURCE_DIR=${__INSTALL_SH_DIR}/psc/ \
+    TUTORIAL_REPO_DIR=${__INSTALL_SH_DIR} \
+    ${__INSTALL_SH_DIR}/lib/mo \
+        ${__INSTALL_SH_DIR}/psc/jupyter/template/kernel-helper.sh \
+        > ${kernel_dir}/kernel-helper.sh
+
+    chmod u+x ${kernel_dir}/kernel-helper.sh
+    cp ${__INSTALL_SH_DIR}/psc/jupyter/template/logo-32x32.png ${kernel_dir}
+    cp ${__INSTALL_SH_DIR}/psc/jupyter/template/logo-64x64.png ${kernel_dir}
 done
 
+# echo "Generating 'activate.sh' and 'deactivate.sh' in: ${__INSTALL_SH_DIR}" 
+# 
+# TUTORIAL_REPO_DIR=${__INSTALL_SH_DIR} \
+# ${__INSTALL_SH_DIR}/lib/mo \
+#     ${__INSTALL_SH_DIR}/psc/template/activate.sh \
+#     > ${__INSTALL_SH_DIR}/activate.sh
+# 
+# TUTORIAL_REPO_DIR=${__INSTALL_SH_DIR} \
+# ${__INSTALL_SH_DIR}/lib/mo \
+#     ${__INSTALL_SH_DIR}/psc/template/deactivate.sh \
+#     > ${__INSTALL_SH_DIR}/deactivate.sh
+# 
+# echo "Instatiating Julia environment at: ${__INSTALL_SH_DIR}" 
+# 
+# source ${__INSTALL_SH_DIR}/activate.sh
+# julia -e "import Pkg; Pkg.instantiate()"
+
+echo "Done"

lib/get_ext.sh

@@ -0,0 +1,20 @@
+#!/usr/bin/env sh
+set -eu
+
+__GET_EXT_SH_DIR=$(
+    cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd
+)
+
+pushd $__GET_EXT_SH_DIR
+
+# Download
+curl -sSL \
+    https://raw.githubusercontent.com/tests-always-included/mo/master/mo -o mo
+
+# Make executable
+chmod +x mo
+
+# Test
+echo "works" | ./mo
+
+popd