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

Jump to bottom

[Feature Request]: Add Ascend NPU npu_fusion_attention to accelerate training #16199

Open

1 task done

kevin19891229 opened this issue Jul 12, 2024 · 2 comments

Labels

kevin19891229 commented Jul 12, 2024

Is there an existing issue for this?

I have searched the existing issues and checked the recent builds/commits

What would your feature do ?

a simple description of npu_fusion_attention operator
add Ascend NPU npu_fusion_attention to accelerate training

Proposed workflow

Go to add description of npu_fusion_attention operator
add Ascend NPU npu_fusion_attention to accelerate training

Additional information

No response

kevin19891229 added the enhancement label

Author

kevin19891229 commented Jul 12, 2024

Coloured-glaze commented Jul 24, 2024

Please use the following function to achieve the same effect

torch.nn.functional.scaled_dot_product_attention()

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment