Skip to content

test: MambaConfig and varlen helper regression coverage - #1009

Open
Chessing234 wants to merge 3 commits into
state-spaces:mainfrom
Chessing234:test/config-and-model-utils
Open

test: MambaConfig and varlen helper regression coverage#1009
Chessing234 wants to merge 3 commits into
state-spaces:mainfrom
Chessing234:test/config-and-model-utils

Conversation

@Chessing234

Copy link
Copy Markdown
Contributor

Summary

  • Add tests ensuring MambaConfig mutable defaults are independent and JSON round-trip safe.
  • Document vocab padding rounding used by MambaLMHeadModel.
  • Add CUDA regression test keeping cu_seqlens int32 after cumsum for varlen Mamba-3 paths.

Test plan

  • pytest tests/test_config_mamba.py tests/test_varlen_helpers.py

Made with Cursor

Ensure mutable dataclass fields are not shared across instances and that
configs survive json encode/decode for save_pretrained compatibility.
Regression guard for varlen Mamba-3 paths where int32 cu_seqlens are
required by TileLang kernels.
Cover the pad_vocab_size_multiple rounding applied during LM head construction.
@Chessing234
Chessing234 force-pushed the test/config-and-model-utils branch from 090e4cc to 9ed887a Compare August 5, 2026 10:23
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant