Skip to content

Prepare Blockwise + IO infrastructure#7281

Merged
jrbourbeau merged 121 commits intodask:mainfrom
rjzamora:blockwise-rewrite-prelim
Mar 12, 2021
Merged

Prepare Blockwise + IO infrastructure#7281
jrbourbeau merged 121 commits intodask:mainfrom
rjzamora:blockwise-rewrite-prelim

Conversation

@rjzamora
Copy link
Member

This PR is an optional prerequisite for #7042. In other words, this includes a small subset of the changes introduced in that PR, and is intended to simplify the review process.

Changes included here:

  1. Introduces the io_deps input/attribute to a Blockwise layer. This attribute is needed to materialize a collections within a Blockwise layer (i.e. IO).
  2. Moves dask.array's zeros, ones, and full over to Blockwise.
  3. Adds distributed test coverage for array creation with zeros, ones, and full.

@rjzamora
Copy link
Member Author

rjzamora commented Mar 2, 2021

Thanks you for the review @crusaderky - Your comments/suggestions are very valuable!

I am still unsure about the current blockwise_io_reg registration approach. I will try my best to think of something a bit cleaner.

@rjzamora
Copy link
Member Author

rjzamora commented Mar 5, 2021

Oops - looks like #7312 just added a new clone method to Blockwise. Just added a fix to make sure io_deps are passed through when a layer is cloned.

Base automatically changed from master to main March 8, 2021 20:20
@rjzamora
Copy link
Member Author

@madsbk - Not sure if you have time for a quick look here, but your thoughts are very welcome. I am hoping to get this merged very soon :)

Copy link
Member

@jrbourbeau jrbourbeau left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for your patience on this PR @rjzamora, this is in

@jrbourbeau jrbourbeau merged commit 3763da9 into dask:main Mar 12, 2021
douglasdavis pushed a commit to douglasdavis/dask that referenced this pull request Mar 14, 2021
@ian-r-rose ian-r-rose mentioned this pull request Mar 19, 2021
2 tasks
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.

4 participants