Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
37 changes: 29 additions & 8 deletions .azure-pipelines/azure-pipelines-linux.yml

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Original file line number Diff line number Diff line change
@@ -1,3 +1,7 @@
c_compiler:
- gcc
c_compiler_version:
- '13'
c_stdlib:
- sysroot
c_stdlib_version:
Expand Down Expand Up @@ -27,6 +31,7 @@ python:
target_platform:
- linux-64
zip_keys:
- - cxx_compiler_version
- - c_compiler_version
- cxx_compiler_version
- c_stdlib_version
- cuda_compiler_version
Original file line number Diff line number Diff line change
@@ -1,3 +1,7 @@
c_compiler:
- gcc
c_compiler_version:
- '13'
c_stdlib:
- sysroot
c_stdlib_version:
Expand Down Expand Up @@ -27,6 +31,7 @@ python:
target_platform:
- linux-64
zip_keys:
- - cxx_compiler_version
- - c_compiler_version
- cxx_compiler_version
- c_stdlib_version
- cuda_compiler_version
Original file line number Diff line number Diff line change
@@ -1,3 +1,7 @@
c_compiler:
- gcc
c_compiler_version:
- '13'
c_stdlib:
- sysroot
c_stdlib_version:
Expand Down Expand Up @@ -27,6 +31,7 @@ python:
target_platform:
- linux-64
zip_keys:
- - cxx_compiler_version
- - c_compiler_version
- cxx_compiler_version
- c_stdlib_version
- cuda_compiler_version
Original file line number Diff line number Diff line change
@@ -1,3 +1,7 @@
c_compiler:
- gcc
c_compiler_version:
- '13'
c_stdlib:
- sysroot
c_stdlib_version:
Expand Down Expand Up @@ -27,6 +31,7 @@ python:
target_platform:
- linux-64
zip_keys:
- - cxx_compiler_version
- - c_compiler_version
- cxx_compiler_version
- c_stdlib_version
- cuda_compiler_version
2 changes: 1 addition & 1 deletion .github/CODEOWNERS

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

24 changes: 10 additions & 14 deletions .scripts/build_steps.sh

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

17 changes: 9 additions & 8 deletions README.md

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

3 changes: 3 additions & 0 deletions conda-forge.yml
Original file line number Diff line number Diff line change
@@ -1,6 +1,9 @@
azure:
free_disk_space: true
github:
branch_name: main
tooling_branch_name: main
conda_build:
error_overlinking: true
conda_build_tool: rattler-build
conda_forge_output_validation: true
64 changes: 0 additions & 64 deletions recipe/meta.yaml

This file was deleted.

74 changes: 74 additions & 0 deletions recipe/recipe.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,74 @@
schema_version: 1

context:
name: mamba-ssm
version: 2.2.5

package:
name: ${{ name|lower }}
version: ${{ version }}

source:
url: https://github.com/state-spaces/mamba/archive/refs/tags/v${{ version }}.tar.gz
sha256: 1e072841619af9c171709b2c6ff17daa0d9bf7caf0fcb083f5c11754b63c6cdb

build:
number: 1
script:
env:
MAMBA_FORCE_BUILD: "TRUE"
content:
- sed -i.bak 's@"ninja"@#"ninja"@g' setup.py
- sed -i.bak 's@"ninja"@#"ninja"@g' pyproject.toml
- ${{ PYTHON }} -m pip install . -vv --no-deps --no-build-isolation
skip:
# mamba-ssm requires CUDA (or ROCm) GPU
- cuda_compiler_version == "None" or not linux

requirements:
build:
- ${{ compiler('c') }}
- ${{ compiler('cxx') }}
- ${{ compiler('cuda') }}
- ${{ stdlib('c') }}
host:
- cuda-version ==${{ cuda_compiler_version }}
- ninja
- packaging
- pip
- python
- pytorch * cuda*
- setuptools >=61.0.0
- triton
- wheel
run:
- causal-conv1d >=1.5.2
- einops
- packaging
- python
- pytorch
- setuptools >=61.0.0
- transformers

tests:
# Need a GPU to import mamba_ssm, can re-enable when pytorch depends on triton>=3.4.0
# containing the https://github.com/triton-lang/triton/pull/6570 bugfix
# - python:
# imports:
# - mamba_ssm
- requirements:
run:
- pip
script:
- pip check

about:
homepage: https://github.com/state-spaces/mamba
summary: Mamba state-space model
license: Apache-2.0
license_file: LICENSE

extra:
recipe-maintainers:
- timkpaine
- weiji14
Copy link
Copy Markdown
Member Author

@weiji14 weiji14 Sep 10, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Added myself to the recipe maintainer list if you don't mind! I'd be interested in adding aarch builds later for Grace Hopper systems (xref conda-forge/conda-forge-pinning-feedstock#7764). Can also help with the CUDA 12.9 migration at #1 later.