Skip to content

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

Draft
MagicalMogu wants to merge 1 commit intomasterfrom
dropout_mask_npu_support
Draft

Modify dropout mask tensor shape and bits format for NPU#10620
MagicalMogu wants to merge 1 commit intomasterfrom
dropout_mask_npu_support

Conversation

@MagicalMogu
Copy link
Contributor

@MagicalMogu MagicalMogu commented Mar 5, 2025

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

@CLAassistant
Copy link

CLAassistant commented Mar 5, 2025

CLA assistant check
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

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants