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

Inplace Blockwise outputs on inputs in Numba backend #1030

Open
ricardoV94 opened this issue Oct 11, 2024 · 0 comments
Open

Inplace Blockwise outputs on inputs in Numba backend #1030

ricardoV94 opened this issue Oct 11, 2024 · 0 comments

Comments

@ricardoV94
Copy link
Member

ricardoV94 commented Oct 11, 2024

Description

When we have a Blockwise with the same output shape and type as one of its inputs (such as Cholesky whose signature is "(m,m)->(m,m)") we can overwrite the output in the input buffer.

This is independent of allowing the core_op to destroy the input as done in #1028

This is only relevant for the numba backend after #1015 since the other backends don't give us inplace control and numpy vectorize (used in the Python backend) neither.

@ricardoV94 ricardoV94 changed the title Inplace Blockwise outputs on inputs Inplace Blockwise outputs on inputs in Numba backend Oct 11, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

1 participant