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

Add torch._foreach_ for speed up #287

Open
sdbds opened this issue Nov 5, 2024 · 0 comments
Open

Add torch._foreach_ for speed up #287

sdbds opened this issue Nov 5, 2024 · 0 comments
Assignees
Labels
feature request Request features

Comments

@sdbds
Copy link
Contributor

sdbds commented Nov 5, 2024

like use
torch._foreach_mul_
torch._foreach_add_
torch._foreach_addcmul_
torch._foreach_sqrt
torch._foreach_zeros_like

@sdbds sdbds added the feature request Request features label Nov 5, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature request Request features
Projects
None yet
Development

No branches or pull requests

2 participants