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.

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

lower NativeDropoutBackward (#5642) and lower native_dropout (#5643) #5647

Merged
merged 2 commits into from
Sep 26, 2023

Conversation

ManfeiBai
Copy link
Collaborator

@ManfeiBai ManfeiBai commented Sep 26, 2023

cherry-pick #5642 and #5643

  • lower NativeDropoutBackward

  • fix lowering and add python test

  • Fix OP lowering for op native_dropout #5588.

  • Added the manual lowering for native_dropout op. We need to add the test to verify whether the logic is correct or not.

JackCaoG and others added 2 commits September 26, 2023 17:13
* lower NativeDropoutBackward

* fix lowering and add python test
* prototype version (compiling error)

* Add native_dropout manual lowering.

* fix to tensor IR and add a simple native_dropout test

* fix data type issue and update test case

* fix IR hash issue

* fix corner case when probability==0

* remove typo line

* add test case when probability=0
@ManfeiBai ManfeiBai changed the title lower NativeDropoutBackward (#5642) lower NativeDropoutBackward (#5642) and lower native_dropout (#5643) Sep 26, 2023
@ManfeiBai ManfeiBai marked this pull request as ready for review September 26, 2023 17:22
@ManfeiBai ManfeiBai merged commit b840137 into r2.1 Sep 26, 2023
7 checks passed
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