Adding more environment specific configs

ci/setup_spack.sh

@@ -26,11 +26,24 @@ then
   spack/bin/spack env create ${environment}
 fi
 
-for file in mirrors.yaml packages.yaml concretizer.yaml config.yaml spack.yaml packages_stack.yaml
+for file in mirrors packages concretizer config spack
 do
-  if [ -e "${STACK_CONFIG_PATH}/${file}" ]; then
-    echo "Copying ${STACK_CONFIG_PATH}/${file} to ${SPACK_SYSTEM_CONFIG_PATH}"
-    cp "${STACK_CONFIG_PATH}/${file}" ${SPACK_SYSTEM_CONFIG_PATH}
+  if [ -e "${STACK_CONFIG_PATH}/${file}.yaml" ]; then
+    echo "Copying ${STACK_CONFIG_PATH}/${file}.yaml to ${SPACK_SYSTEM_CONFIG_PATH}"
+    cp "${STACK_CONFIG_PATH}/${file}.yaml" \
+      ${SPACK_SYSTEM_CONFIG_PATH}
+  fi
+
+  if [ -e "${STACK_CONFIG_PATH}/${file}_stack.yaml" ]; then
+    echo "Copying ${STACK_CONFIG_PATH}/${file}_stack.yaml to ${SPACK_SYSTEM_CONFIG_PATH}"
+    cp "${STACK_CONFIG_PATH}/${file}_stack.yaml" \
+      ${SPACK_SYSTEM_CONFIG_PATH}
+  fi
+
+  if [ -e "${STACK_CONFIG_PATH}/${file}_${environment}.yaml" ]; then
+    echo "Copying ${STACK_CONFIG_PATH}/${file}_${environment}.yaml to ${SPACK_SYSTEM_CONFIG_PATH}/${file}_env.yaml"
+    cp "${STACK_CONFIG_PATH}/${file}_${environment}.yaml" \
+      ${SPACK_SYSTEM_CONFIG_PATH}/${file}_env.yaml
   fi
 done
 
@@ -109,8 +122,6 @@ spack/bin/spack buildcache keys \
 
 #spack/bin/spack buildcache update-index /buildcache
 
-cp ${STACK_CONFIG_PATH}/packages_${environment}.yaml ${SPACK_SYSTEM_CONFIG_PATH}/packages_env.yaml
-
 echo "packages:" > ${SPACK_SYSTEM_CONFIG_PATH}/packages.yaml
 export system_compiler=$(jq -Mrc '
 .stack

stacks/pinot-noir/definitions_jed.yaml

+definitions:
+  - cuda_system_codes:
+      - cuda
+      - cudnn
+
+  - cuda_serial_codes:
+      - nccl

stacks/pinot-noir/spack.yaml

@@ -4,6 +4,7 @@ spack:
       - packages_stack.yaml
       - packages_env.yaml
       - modules_stack.yaml
+      - definitions_env.yaml
 
   definitions:
     # -------------------------------------------------------------------------
@@ -89,6 +90,11 @@ spack:
         - superlu
         - voropp
 
+    # -------------------------------------------------------------------------
+    # Empty to specialize in environments
+    # -------------------------------------------------------------------------
+    - cuda_system_codes: []
+
     # -------------------------------------------------------------------------
     # Parallel codes
     # -------------------------------------------------------------------------