Skip to content

Prepare drop pack unpack - 2nd attempt#10041

Closed
fjetter wants to merge 2 commits intodask:mainfrom
fjetter:prepare_pack_unpack_again
Closed

Prepare drop pack unpack - 2nd attempt#10041
fjetter wants to merge 2 commits intodask:mainfrom
fjetter:prepare_pack_unpack_again

Conversation

@fjetter
Copy link
Copy Markdown
Member

@fjetter fjetter commented Mar 9, 2023

So... #9994 was not my brightest moment. While I cached the self._split_keys in SimpleShuffleLayer I forgot that the key in self._split_keys thing is a linear time lookup 🤦‍♂️ causing graph generation to be abysmally slow.

This could be easily verified with micro benchmarks. By using a set here we're back to normal

@fjetter
Copy link
Copy Markdown
Member Author

fjetter commented Mar 10, 2023

It might not be a good idea after all to merge this while the pack/unpack protocol still lives since the callable annotation triggers a graph materialization on client side...

@fjetter fjetter closed this Mar 10, 2023
@fjetter fjetter mentioned this pull request Mar 21, 2023
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