state-spaces/mamba#771 GitHub Actions CI using EC2 GPU nodes
This PR adds 4 GitHub Actions workflows that install or test mamba_ssm
on EC2 GPU nodes (using [Open-Athena/ec2-gha]):
install.yaml
: installmamba_ssm
on an EC2 GPU instance (default: [g4dn.xlarge
])installs.yaml
: runinstall.yaml
on 6 recent versions of Mamba (2.2.{0,1,2,3post2,4,5}
)test.yaml
: runmamba_ssm
tests on an EC2 GPU instance ([g5
] or [g6
] series)tests.yaml
: runtest.yaml
on HEAD, on ag5.2xlarge
andg6.2xlarge