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.

Sign up for GitHub

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

[ROCm] Improve softmax performance. #1740

Open
wants to merge 2 commits into
base: release/2.4
Choose a base branch
from
+195 −35

Fix fast softmax for non-ILP multiple sizes.

65d155c
Select commit
Loading
Failed to load commit list.
Open

[ROCm] Improve softmax performance. #1740

Fix fast softmax for non-ILP multiple sizes.
65d155c
Select commit
Loading
Failed to load commit list.
ROCm Repo Management API / Tests / Tests / Test Distributed / Run pytorch_distributed_2 succeeded Jan 28, 2025 in 0s

skipped: 329, passed: 491

Send us feedback

Details

View more details on ROCm Repo Management API