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

[ExportedProgram] Add mutable attribute to buffer #123

Conversation

chrsmcgrr
Copy link
Contributor

Fixes #85

PR based on the work of @maxbartel

Requires changes in torch-mlir: llvm/torch-mlir/#3688

Adds the mutable modifier to a global buffer and lifts said buffer to a global if there is a store-producer node associated with it.

@chrsmcgrr chrsmcgrr force-pushed the mcgirr/roo-43-fix-kv-cache-issue-for-torch-buffers branch 2 times, most recently from ed09e6a to bbc098d Compare September 10, 2024 09:33
@chrsmcgrr
Copy link
Contributor Author

chrsmcgrr commented Sep 18, 2024

Requires newer version of IREE containing the torch-mlir change: #151

@chrsmcgrr chrsmcgrr force-pushed the mcgirr/roo-43-fix-kv-cache-issue-for-torch-buffers branch from 735c209 to 2a7058f Compare September 19, 2024 10:22
@chrsmcgrr
Copy link
Contributor Author

Requires newer version of IREE containing the torch-mlir change: #151

Actually this PR did not have the IREE changes that were required. A newer release is needed.

maxbartel pushed a commit that referenced this pull request Sep 20, 2024
ci(iree): update version

to release 20240918.1020

This update is required for another PR
#123

Signed-off-by: Christopher McGirr <[email protected]>
@maxbartel
Copy link
Contributor

Could you rebase, please? The tests should now pass

@chrsmcgrr chrsmcgrr force-pushed the mcgirr/roo-43-fix-kv-cache-issue-for-torch-buffers branch from 2a7058f to d07db5f Compare September 23, 2024 06:53
@maxbartel
Copy link
Contributor

@stellaraccident This seems to work now with the newest IREE bump. Can we get a review from you please?

Copy link
Collaborator

@stellaraccident stellaraccident left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Amazing -- I don't know if you truly know how many things had to come together to make that work. I've been slowly making progress on that functionality for the past year because it seemed very useful to get mapped right, but I didn't know we were so close to having it fully working.

@stellaraccident stellaraccident merged commit 4e95351 into iree-org:main Sep 24, 2024
8 checks passed
IanNod pushed a commit to IanNod/iree-turbine that referenced this pull request Sep 30, 2024
ci(iree): update version

to release 20240918.1020

This update is required for another PR
iree-org#123

Signed-off-by: Christopher McGirr <[email protected]>
IanNod pushed a commit to IanNod/iree-turbine that referenced this pull request Sep 30, 2024
Fixes iree-org#85

PR based on the work of @maxbartel 

Requires changes in torch-mlir:
[llvm/torch-mlir/#3688](llvm/torch-mlir#3688)

Adds the mutable modifier to a global buffer and lifts said buffer to a
global if there is a store-producer node associated with it.

Signed-off-by: Christopher McGirr <[email protected]>
Co-authored-by: Maximilian Bartel <[email protected]>
IanNod pushed a commit to IanNod/iree-turbine that referenced this pull request Sep 30, 2024
ci(iree): update version

to release 20240918.1020

This update is required for another PR
iree-org#123

Signed-off-by: Christopher McGirr <[email protected]>
Signed-off-by: Ian <[email protected]>
IanNod pushed a commit to IanNod/iree-turbine that referenced this pull request Sep 30, 2024
Fixes iree-org#85

PR based on the work of @maxbartel

Requires changes in torch-mlir:
[llvm/torch-mlir/#3688](llvm/torch-mlir#3688)

Adds the mutable modifier to a global buffer and lifts said buffer to a
global if there is a store-producer node associated with it.

Signed-off-by: Christopher McGirr <[email protected]>
Co-authored-by: Maximilian Bartel <[email protected]>
Signed-off-by: Ian <[email protected]>
IanNod pushed a commit to IanNod/iree-turbine that referenced this pull request Sep 30, 2024
ci(iree): update version

to release 20240918.1020

This update is required for another PR
iree-org#123

Signed-off-by: Christopher McGirr <[email protected]>
Signed-off-by: Ian <[email protected]>
IanNod pushed a commit to IanNod/iree-turbine that referenced this pull request Sep 30, 2024
Fixes iree-org#85

PR based on the work of @maxbartel

Requires changes in torch-mlir:
[llvm/torch-mlir/#3688](llvm/torch-mlir#3688)

Adds the mutable modifier to a global buffer and lifts said buffer to a
global if there is a store-producer node associated with it.

Signed-off-by: Christopher McGirr <[email protected]>
Co-authored-by: Maximilian Bartel <[email protected]>
Signed-off-by: Ian <[email protected]>
stellaraccident pushed a commit that referenced this pull request Oct 13, 2024
Fixes #85

PR based on the work of @maxbartel 

Requires changes in torch-mlir:
[llvm/torch-mlir/#3688](llvm/torch-mlir#3688)

Adds the mutable modifier to a global buffer and lifts said buffer to a
global if there is a store-producer node associated with it.

Signed-off-by: Christopher McGirr <[email protected]>
Co-authored-by: Maximilian Bartel <[email protected]>
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.

Fail to compile simple kv_cache
3 participants