Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[i1] Implement packed_storage layout encoding attribute #19354

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

lialan
Copy link
Contributor

@lialan lialan commented Dec 3, 2024

  • make packed_storage as a type of iree_encoding attribute, and make type converters accept it.
  • i1 tensors with #iree_encoding.packed_storage will be interpreted as packed i1 type, same as specifying --iree-experimental-packed-i1-storage. Other i1 tensors are treated as non-packed datatype, and will be extended.
  • --iree-experimental-packed-i1-storage are kept for testing purposes.
    • We can drop this option after frontend enables emitting i1 tensors with attributes.

@lialan lialan self-assigned this Dec 3, 2024
@lialan lialan linked an issue Dec 3, 2024 that may be closed by this pull request
@lialan lialan changed the title Implement i1_packed_storage layout encoding attribute Implement packed_storage layout encoding attribute Dec 5, 2024
@lialan lialan force-pushed the lialan/i1_attr branch 3 times, most recently from 1f2e200 to 8588cfd Compare December 13, 2024 03:37
@lialan lialan marked this pull request as ready for review December 13, 2024 05:28
* make `packed_storage` as a type of `iree_encoding` attribute, and make type converters accept it.
* `i1` tensors with `#iree_encoding.packed_storage` will be interpreted as packed i1 type, same as specifying `--iree-experimental-packed-i1-storage`.
* `--iree-experimental-packed-i1-storage` are kept for testing purposes. We can drop this option after frontend enables emitting `i1` tensors with attributes.

Signed-off-by: Alan Li <[email protected]>
@lialan lialan changed the title Implement packed_storage layout encoding attribute [i1] Implement packed_storage layout encoding attribute Dec 13, 2024
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] Support packed i1 storage through encodings
1 participant