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.

Sign up for GitHub

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

Jump to bottom

Add Numba implementation of Blockwise #1015

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Add Numba implementation of Blockwise #1015

wants to merge 2 commits into from

Commits on Oct 7, 2024

  1. Use infer_shape of core_op to infer Blockwise core shapes 

    This can only be done when the output of infer_shape of the core_op depends only on the input shapes, and not their values.
    @ricardoV94
    ricardoV94 committed Oct 7, 2024
    Configuration menu
    Copy the full SHA
    1e06295 View commit details
    Browse the repository at this point in the history

  2. Add Numba implementation of Blockwise 

    Restricted to 3 outputs, due to limitations in jitting of Numba functions
    @ricardoV94
    ricardoV94 committed Oct 7, 2024
    Configuration menu
    Copy the full SHA
    31cc1e9 View commit details
    Browse the repository at this point in the history