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

Reland perf fix for nan inf check #7184

Open
wants to merge 6 commits into
base: master
Choose a base branch
from

Conversation

nelyahu
Copy link
Contributor

@nelyahu nelyahu commented Mar 30, 2025

replace previous usage with logical ops for nan/inf detect with torch.where

@nelyahu nelyahu force-pushed the reland_perf_fix_for_nan_inf_check branch from 6f123cc to 2166ff4 Compare March 30, 2025 08:05
@nelyahu
Copy link
Contributor Author

nelyahu commented Mar 30, 2025

@tjruwase please see fix for the issue in #7171

@nelyahu nelyahu force-pushed the reland_perf_fix_for_nan_inf_check branch from 889c90d to cddef4c Compare March 31, 2025 09:05
@loadams
Copy link
Collaborator

loadams commented Mar 31, 2025

@nelyahu - could you run the pre-commit formatter and then we can merge this?

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