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

Reduce the overhead of using LocalExecutor (esp. dtype validation) #219

Merged
merged 6 commits into from
Feb 14, 2023

Conversation

karlhigley
Copy link
Contributor

No description provided.

We usually want this in order to catch places where operators are producing data that doesn't match their output schemas, but it produces way too much performance overhead when run on every batch in the dataloaders.
We can speed up both code paths here by assuming we're on the happy path and catching errors when they happen instead of calling `hash` first to check.
We don't need the extra logic from `Dtype.with_shape` in this case, since we already know that `Shape()` is a valid shape.
@karlhigley karlhigley added clean up chore Maintenance for the repository labels Feb 13, 2023
@karlhigley karlhigley added this to the Merlin 23.02 milestone Feb 13, 2023
@karlhigley karlhigley self-assigned this Feb 13, 2023
@karlhigley karlhigley merged commit 0f6ee98 into NVIDIA-Merlin:main Feb 14, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
chore Maintenance for the repository clean up
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants