Modify dropout mask tensor shape and bits format for NPU #10620

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

Draft

MagicalMogu wants to merge 1 commit into master from dropout_mask_npu_support

Contributor

MagicalMogu commented Mar 5, 2025 •

edited

Loading

为了实现aclnn dropout，修改了mask的shape为一维且对齐128的bits格式。需要和NPU分支搭配使用，详细见链接。https://github.com/Oneflow-Inc/oneflow-npu/pull/323


          Modify dropout mask tensor shape and bits format for NPU

CLAassistant commented Mar 5, 2025 •

edited

Loading

All committers have signed the CLA.

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

Labels

None yet