add mamba-build (#122)

Required until state-spaces/mamba#729 and state-spaces/mamba#727 are merged and a new mamba release is published on PyPi

Author: dtrifiro

.github/actions/build-mamba/action.yml

@@ -0,0 +1,13 @@
+name: build mamba ssm
+description: "build mamba-ssm"
+runs:
+  using: composite
+  steps:
+  - id: build_mamba
+    name: "build mamba-ssm"
+    run: |
+      ORIGINAL_DIR=$(pwd)
+
+      cd "$CLIENT_PATH"
+      bash ${{ github.action_path }}/build_mamba.sh || (echo "mamba build failed" >&2 && exit 1)
+    shell: bash

.github/actions/build-mamba/build_mamba.sh

@@ -0,0 +1,28 @@
+#!/bin/bash
+# this sets the following github actions env vars:
+#  - whl_file
+#  - sdist_file
+#  - WHL_STATUS
+set -eu
+
+export MAX_JOBS=$(($(nproc) / 2))
+export MAMBA_FORCE_BUILD="TRUE"
+export MAMBA_FORCE_CXX11_ABI="FALSE"
+export HIP_ARCHITECTURES="gfx942" # MI300X, hardcoded for now. See env-build-nm-vllm-ent PYTORCH_ROCM_ARCH
+
+uv build --wheel --sdist || SUCCESS=$?
+SUCCESS=$?
+
+echo "WHL_STATUS=${SUCCESS}" | tee -a "$GITHUB_ENV"
+
+if [ "${SUCCESS}" -ne 0 ]; then
+	exit 1
+fi
+
+MAMBA_WHL="$(find dist -type f -iname "*\.whl" -exec basename {} \;)"
+uv run --pre --no-project --with wheel-metadata-injector wheel-metadata-injector "dist/${MAMBA_WHL}"
+
+MAMBA_TARFILE="$(find dist -type f -iname "*\.tar.gz" -exec basename {} \;)"
+
+echo "WHL_FILE=${MAMBA_WHL}" | tee -a "$GITHUB_ENV"
+echo "SDIST_FILE=${MAMBA_TARFILE}" | tee -a "$GITHUB_ENV"