Skip to content

Avoid unbounded memory usage in matmul by delegating to dot#6924

Closed
tomwhite wants to merge 1 commit intodask:masterfrom
tomwhite:matmul-delegate-to-dot
Closed

Avoid unbounded memory usage in matmul by delegating to dot#6924
tomwhite wants to merge 1 commit intodask:masterfrom
tomwhite:matmul-delegate-to-dot

Conversation

@tomwhite
Copy link
Copy Markdown
Contributor

@tomwhite tomwhite commented Dec 3, 2020

... in the common case of 1D or 2D arrays (#6874)

Note that this depends on #6916 to be fixed to have an effect though.

  • Tests added / passed
  • Passes black dask / flake8 dask

@jakirkham
Copy link
Copy Markdown
Member

This code is actually kind of old. Since Dask added support for dispatching gufuncs directly on Dask Array and IIRC matmul is a gufunc, I wonder if we should just do that instead. Though others please feel to correct me if that is wrong 🙂

@mrocklin
Copy link
Copy Markdown
Member

mrocklin commented Dec 4, 2020 via email

@jakirkham
Copy link
Copy Markdown
Member

Sorry what trick are you referring to?

@tomwhite
Copy link
Copy Markdown
Contributor Author

@jakirkham the trick is described in #6874.

@tomwhite
Copy link
Copy Markdown
Contributor Author

Closing in favour of #7000

@tomwhite tomwhite closed this Jan 11, 2021
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.

3 participants