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

Remove conservative checks for supported Subtensors operations in JAX #849

Merged
merged 1 commit into from
Jun 28, 2024
Merged

Remove conservative checks for supported Subtensors operations in JAX #849

merged 1 commit into from
Jun 28, 2024

Commits on Jun 24, 2024

  1. Remove false positive check for supported Subtensors operations in JAX 

    The check was failing incorrectly for cases that are supported such as constant Boolean arrays.
Besides that, user may dispatch without necessarily jitting the graph. There is no reason to fail eagerly.
    @ricardoV94
    ricardoV94 committed Jun 24, 2024
    Configuration menu
    Copy the full SHA
    2b99224 View commit details
    Browse the repository at this point in the history