Skip to content

Conversation

@zhangboSJTU
Copy link
Collaborator

@zhangboSJTU zhangboSJTU commented Apr 17, 2023

Add test dropout_nd api and configs

results:

dims p axis dtype time
[16, 16, 16] 0.5 [1] fp32 11.8
[16, 16, 16, 16] 0.5 [0,1] fp32 47.7
[32, 108, 768] 0.1 [0] fp32 222.3

@paddle-bot
Copy link

paddle-bot bot commented Apr 17, 2023

Thanks for your contribution! Please wait for the result of CI firstly.



# no dropout_nd in pytorch
def dropout_nd(x,
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

这里为啥要重新定义一遍dropout_nd

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

dropout_nd 的使用需要调用 paddle._legacy_C_ops.dropout_nd, paddle.dropout 走不到这里

return out


@benchmark_registry.register("dropout_nd")
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

好像少个config模块

Copy link
Collaborator Author

@zhangboSJTU zhangboSJTU Sep 22, 2023

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

done,make run_torch false

@Xreki Xreki merged commit bd98227 into PaddlePaddle:master Sep 22, 2023
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.

3 participants