Skip to content
This repository was archived by the owner on Jan 28, 2026. It is now read-only.

Add torch.rsqrt check#100

Open
cyyever wants to merge 2 commits intometa-pytorch:mainfrom
cyyever:check_rsqrt
Open

Add torch.rsqrt check#100
cyyever wants to merge 2 commits intometa-pytorch:mainfrom
cyyever:check_rsqrt

Conversation

@cyyever
Copy link
Copy Markdown

@cyyever cyyever commented Sep 14, 2025

Suggest using a*torch.rsqrt(b) because it is faster than a/torch.sqrt(b) if the accelerator provides the instructions.
See https://docs.pytorch.org/docs/stable/generated/torch.rsqrt.html

@meta-cla meta-cla Bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Sep 14, 2025
Signed-off-by: Yuanyuan Chen <cyyever@outlook.com>
Signed-off-by: Yuanyuan Chen <cyyever@outlook.com>
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants