better pyenv install for ellis

marchdf · Feb 6, 2024 · 1d9c7fe · 1d9c7fe
1 parent 048d4f5
commit 1d9c7fe
Showing 1 changed file with 30 additions and 17 deletions.
diff --git a/dot_local/share/zinit/plugins/hpc/hpc.plugin.zsh.tmpl b/dot_local/share/zinit/plugins/hpc/hpc.plugin.zsh.tmpl
@@ -45,7 +45,6 @@ function load_custom_modules {
     {
         module purge
         module load gcc
-        module load mpt
         module load binutils
         module load bzip2
         module load cmake
@@ -56,12 +55,15 @@ function load_custom_modules {
         module load image-magick
         module load libffi
         module load libtool
-        module load openssl
         module load python/3.7.7
         module load py-setuptools-scm
         module load sqlite
-        module load texlive
         module load tmux
+        if [ "${NREL_CLUSTER}" = "eagle" ]; then
+            module load mpt
+            module load openssl
+            module load texlive
+        fi
 
         module load emacs/28.2
         module load ccls
@@ -82,20 +84,31 @@ function pvtunnel() {
     ssh -L 11111:$1:11111 mhenryde@$1
 }
 
-function pyenv_python_install(){
-    module load gcc
-    module load mpt
-    module load binutils
-    module load git
-    module load gnutls
-    module load libtool
-    module load texlive
-    module load bzip2
-    module load libffi
-    module load openssl
-    module load sqlite
-    export LDFLAGS="-L${OPENSSL_ROOT_DIR}/lib -L${LIBFFI_ROOT_DIR}/lib64 -L${BZIP2_ROOT_DIR}/lib -L${SQLITE_ROOT_DIR}/lib"
-    export CPPFLAGS="-I${OPENSSL_ROOT_DIR}/include -I${LIBFFI_ROOT_DIR}/include -I${BZIP2_ROOT_DIR}/include -I${SQLITE_ROOT_DIR}/include"
+function pyenv_python_install() {
+    if [ "${NREL_CLUSTER}" = "eagle" ]; then
+        module load gcc
+        module load binutils
+        module load git
+        module load gnutls
+        module load libtool
+        module load bzip2
+        module load libffi
+        module load openssl
+        module load sqlite
+        export LDFLAGS="-L${OPENSSL_ROOT_DIR}/lib -L${LIBFFI_ROOT_DIR}/lib64 -L${BZIP2_ROOT_DIR}/lib -L${SQLITE_ROOT_DIR}/lib"
+        export CPPFLAGS="-I${OPENSSL_ROOT_DIR}/include -I${LIBFFI_ROOT_DIR}/include -I${BZIP2_ROOT_DIR}/include -I${SQLITE_ROOT_DIR}/include"
+    elif [ "${NREL_CLUSTER}" = "ellis" ]; then
+        module load gcc
+        module load binutils
+        module load git
+        module load gnutls
+        module load libtool
+        module load bzip2
+        module load libffi
+        module load sqlite
+        export LDFLAGS="-L${LIBFFI_ROOT_DIR}/lib64 -L${BZIP2_ROOT_DIR}/lib -L${SQLITE_ROOT_DIR}/lib"
+        export CPPFLAGS="-I${LIBFFI_ROOT_DIR}/include -I${BZIP2_ROOT_DIR}/include -I${SQLITE_ROOT_DIR}/include"
+    fi
     export PYTHON_CONFIGURE_OPTS="--enable-optimizations --with-lto"
     export PYTHON_CFLAGS="-mtune=native"
     pyenv install