Skip to content

AOTI Cuda Partitioner Should Not Take Method Name as an arg #14910

@JacobSzwejbka

Description

@JacobSzwejbka

🚀 The feature, motivation and pitch

This doesnt seem like it should be the users responsibility to manage. Ideally the delegate would name its blobs on its own and know which one its looking for, this is what we expect of other backends when putting blobs in the named data map. Because it takes it as an arg it also creates a weird pattern where we have to create a separate partitioner instance for each entry point when we should really be able to use the same one across all of them.

example usage huggingface/optimum-executorch@dfa2d7a

Alternatives

No response

Additional context

No response

RFC (Optional)

No response

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

Status

To triage

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions