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

Make the new multiple-inputs optimization function the default. #412

Merged
merged 3 commits into from
Mar 8, 2024

Conversation

tomwhite
Copy link
Member

@tomwhite tomwhite commented Mar 5, 2024

I've been testing this a bit at scale, and I think it's ready to enable the new optimization function (multiple_inputs_optimize_dag) as the default. (#397 explains how it works in more detail.)

@tomwhite
Copy link
Member Author

tomwhite commented Mar 6, 2024

The test_expand_dims failure in the array API tests has found a real bug which I'm looking into.

@tomwhite tomwhite force-pushed the enable-new-optimizer-by-default branch from 247d52e to 31e7eb7 Compare March 8, 2024 12:46
@tomwhite
Copy link
Member Author

tomwhite commented Mar 8, 2024

The test_expand_dims failure in the array API tests has found a real bug which I'm looking into.

Fixed in #416

@tomwhite
Copy link
Member Author

tomwhite commented Mar 8, 2024

This also enables the new reduction implementation by default. It's possible to disable on a per-function basis if needed by calling e.g. xp.mean(x, use_new_impl=False).

@tomwhite tomwhite merged commit 9dd82ad into main Mar 8, 2024
7 checks passed
@tomwhite tomwhite deleted the enable-new-optimizer-by-default branch March 8, 2024 17:07
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.

1 participant