Skip to content

✨[Feature] Add min_block_size exemption for fully-supported models #2131

Closed
@gs-olive

Description

@gs-olive

Context

Minimum block size can be disregarded in the partitioner if the entire model is end-to-end supported.

Proposal

Add a special case to the partitioner for end-to-end support and automatically enable any block size in that scenario.

Metadata

Metadata

Assignees

Labels

Story: Export/Compile UnificationIssues relating to unification of Dynamo compile/export pathscomponent: dynamoIssues relating to the `torch.compile` or `torch._dynamo.export` pathsfeature requestNew feature or request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions