e4s ci: add py-mpi4py, py-numba (#42845)

share/spack/gitlab/cloud_pipelines/stacks/e4s-neoverse-v2/spack.yaml

@@ -188,7 +188,9 @@ spack:
   - py-jax
   - py-jupyterlab
   - py-matplotlib
+  - py-mpi4py
   - py-notebook
+  - py-numba
   - py-numpy
   - py-openai
   - py-pandas

share/spack/gitlab/cloud_pipelines/stacks/e4s-neoverse_v1/spack.yaml

@@ -180,6 +180,27 @@ spack:
   # - libpressio +bitgrooming +bzip2 ~cuda ~cusz +fpzip +hdf5 +libdistributed +lua +openmp +python +sz +sz3 +unix +zfp # py-numcodecs@0.7.3: gcc: error: unrecognized command-line option '-mno-sse2'
   # - variorum                      # variorum: https://github.com/spack/spack/issues/38786
 
+  # PYTHON PACKAGES
+  - opencv +python3
+  - py-horovod
+  - py-jax
+  - py-jupyterlab
+  - py-matplotlib
+  - py-mpi4py
+  - py-notebook
+  - py-numba
+  - py-numpy
+  - py-openai
+  - py-pandas
+  - py-plotly
+  - py-pooch
+  - py-pytest
+  - py-scikit-learn
+  - py-scipy
+  - py-seaborn
+  - py-tensorflow
+  - py-torch
+
   # CUDA NOARCH
   - flux-core +cuda
   - hpctoolkit +cuda
@@ -281,25 +302,6 @@ spack:
   # - slepc +cuda cuda_arch=80      # slepc: make[1]: *** internal error: invalid --jobserver-auth string 'fifo:/tmp/GMfifo1313'.
   # - upcxx +cuda cuda_arch=80      # upcxx: needs NVIDIA driver
 
-  # PYTHON PACKAGES
-  - opencv +python3
-  - py-horovod
-  - py-jax
-  - py-jupyterlab
-  - py-matplotlib
-  - py-notebook
-  - py-numpy
-  - py-openai
-  - py-pandas
-  - py-plotly
-  - py-pooch
-  - py-pytest
-  - py-scikit-learn
-  - py-scipy
-  - py-seaborn
-  - py-tensorflow
-  - py-torch
-
   # CUDA 90
   - amrex +cuda cuda_arch=90
   - arborx +cuda cuda_arch=90 ^kokkos +wrapper

share/spack/gitlab/cloud_pipelines/stacks/e4s-oneapi/spack.yaml

@@ -199,10 +199,7 @@ spack:
 
   # PYTHON PACKAGES
   - opencv +python3
-  # - py-horovod # error
-  # - py-jax # error
   - py-jupyterlab
-  # - py-matplotlib #  error
   - py-notebook
   - py-numpy
   - py-openai
@@ -213,8 +210,13 @@ spack:
   - py-scikit-learn
   - py-scipy
   - py-seaborn
-  # - py-tensorflow # error
-  # - py-torch # error
+  - py-mpi4py
+  - py-numba
+  # - py-horovod      # error
+  # - py-jax          # error
+  # - py-matplotlib   # error
+  # - py-tensorflow   # error
+  # - py-torch        # error
 
   # GPU
   - amrex +sycl

share/spack/gitlab/cloud_pipelines/stacks/e4s-power/spack.yaml

@@ -185,11 +185,12 @@ spack:
 
   # PYTHON PACKAGES
   - opencv +python3
-  # - py-horovod # py-torch, py-tensorflow
   - py-jax
   - py-jupyterlab
   - py-matplotlib
+  - py-mpi4py
   - py-notebook
+  - py-numba
   - py-numpy
   - py-openai
   - py-pandas
@@ -199,8 +200,9 @@ spack:
   - py-scikit-learn
   - py-scipy
   - py-seaborn
-  # - py-tensorflow # error
-  # - py-torch # error
+  # - py-horovod        # py-torch, py-tensorflow
+  # - py-tensorflow     # error
+  # - py-torch          # error
 
   # CUDA NOARCH
   - bricks +cuda

share/spack/gitlab/cloud_pipelines/stacks/e4s/spack.yaml

@@ -196,7 +196,9 @@ spack:
   - py-jax
   - py-jupyterlab
   - py-matplotlib
+  - py-mpi4py
   - py-notebook
+  - py-numba
   - py-numpy
   - py-openai
   - py-pandas