Try to discover CPU count on our own

Author: milljm

conda/libmesh-vtk/build.sh

@@ -2,6 +2,10 @@
 set -eu
 export PATH=/bin:$PATH
 
+if [ -f utils.sh ]; then
+    source utils.sh
+fi
+
 if [[ $mpi == "openmpi" ]]; then
   export OMPI_MCA_plm=isolated
   export OMPI_MCA_rmaps_base_oversubscribe=yes
@@ -30,8 +34,7 @@ cmake .. -G "Ninja" \
     -DVTK_GROUP_ENABLE_Web:STRING=NO \
     -DCMAKE_OSX_SYSROOT=${CONDA_BUILD_SYSROOT}
 
-CORES=$(echo "${CPU_COUNT:-2} / 2" | bc)
-ninja install -v -j $CORES
+ninja install -v -j $CPU_COUNT
 
 # VTK 9.1 now places libs in lib64 when installed on linux, linking to the "expected" location of lib
 if [[ $(uname) == Linux ]]; then

conda/libmesh-vtk/utils.sh

@@ -0,0 +1 @@
+../mpich/utils.sh

conda/libmesh/build.sh

@@ -10,6 +10,10 @@ if [ -z $PETSC_DIR ]; then
     exit 1
 fi
 
+if [ -f utils.sh ]; then
+    source utils.sh
+fi
+
 function sed_replace(){
     if [ `uname` = "Darwin" ]; then
         sed -i '' -e "s|${BUILD_PREFIX}|${PREFIX}|g" $PREFIX/libmesh/bin/libmesh-config
@@ -79,8 +83,7 @@ LIBMESH_DIR=${PREFIX}/libmesh \
                     --with-vtk-include=${BUILD_PREFIX}/libmesh-vtk/include/vtk-${SHORT_VTK_NAME} \
                     $*
 
-CORES=$(echo "${CPU_COUNT:-2} / 2" | bc)
-make -j $CORES
+make -j $CPU_COUNT
 make install
 sed_replace
 

conda/libmesh/utils.sh

@@ -0,0 +1 @@
+../mpich/utils.sh

conda/mpich/build.sh

@@ -1,13 +1,15 @@
 #!/bin/bash
 set -eu
-printf "\n\n\n\n`env`\n\n\n\n"
 export PATH=/bin:$PATH
-
 export FCFLAGS="$FFLAGS"
 export CC=$(basename "$CC")
 export CXX=$(basename "$CXX")
 export FC=$(basename "$FC")
 
+if [ -f utils.sh ]; then
+    source utils.sh
+fi
+
 if [[ $HOST == arm64-apple-darwin20.0.0 ]]; then
     # use Conda-Forge's Arm64 config.guess and config.sub, see
     # https://conda-forge.org/blog/posts/2020-10-29-macos-arm64/
@@ -92,8 +94,8 @@ fi
             CFLAGS="${CTUNING}" CXXFLAGS="${CTUNING}" FFLAGS="${FTUNING}" LDFLAGS="${LDFLAGS}" \
             FCFLAGS="${FTUNING}" F90FLAGS="" F77FLAGS="" \
             ${OPTIONS}
-CORES=$(echo "${CPU_COUNT:-2} / 2" | bc)
-make -j $CORES
+printf "Building with $CPU_COUNT cores...\n"
+make -j $CPU_COUNT
 make install
 
 # Set MPICH environment variables for those that need it, and set CXXFLAGS using our ACTIVATION_CXXFLAGS variable

conda/mpich/utils.sh

@@ -0,0 +1,10 @@
+# establish cores, or reasonable default
+if [ -z "$CPU_COUNT" ]; then
+    if [ `uname` == "Darwin" ]; then
+        CPU_COUNT=$(echo "$(sysctl -n hw.ncpu) / 2" | bc)
+    elif [ `which nproc` ]; then
+        CPU_COUNT=$(echo `nproc` / 2 | bc)
+    else
+        CPU_COUNT=4
+    fi
+fi

conda/petsc/build.sh

@@ -9,6 +9,10 @@ export CC=$(basename "$CC")
 export CXX=$(basename "$CXX")
 export FC=$(basename "$FC")
 
+if [ -f utils.sh ]; then
+    source utils.sh
+fi
+
 # feed-stock recommendation
 # scrub debug-prefix-map args, which cause problems in pkg-config
 export CFLAGS=$(echo ${CFLAGS:-} | sed -E 's@\-fdebug\-prefix\-map[^ ]*@@g')

conda/petsc/utils.sh

@@ -0,0 +1 @@
+../mpich/utils.sh