Skip to content

[FEATURE] Add slimmable size breakpoint introspection#298

Merged
sdatkinson merged 4 commits into
mainfrom
268-slimmable-breakpoints
Jun 23, 2026
Merged

[FEATURE] Add slimmable size breakpoint introspection#298
sdatkinson merged 4 commits into
mainfrom
268-slimmable-breakpoints

Conversation

@sdatkinson

Copy link
Copy Markdown
Owner

Description

Add SlimmableModel::GetSlimmableSizeBreakpoints() so hosts can enumerate the normalized size-control values where a slimmable model changes size.

Resolves #268.

Summary:

  • Add breakpoint introspection to the shared SlimmableModel interface.
  • Implement breakpoints for SlimmableContainer.
  • Implement breakpoints for SlimmableWavenet, using the sorted unique union of internal ratio dividers across slimmable layer arrays.
  • Use internal divider semantics: 0.0 and 1.0 are implied bounds, so n selectable sizes return n - 1 breakpoints.
  • Add tests for container and WaveNet breakpoint behavior.
  • Leave a TODO to make the new interface method abstract in the next breaking release.

@sdatkinson sdatkinson merged commit 1108b60 into main Jun 23, 2026
4 checks passed
@sdatkinson sdatkinson deleted the 268-slimmable-breakpoints branch June 23, 2026 22:41
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.

Feature Request: a list of slimmable weight breakpoints.

1 participant