Skip to content

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

Merged
ManfeiBai merged 2 commits intor2.1from
r2.1sep25
Sep 26, 2023
Merged

lower NativeDropoutBackward (#5642) and lower native_dropout (#5643)#5647
ManfeiBai merged 2 commits intor2.1from
r2.1sep25

Conversation

@ManfeiBai
Copy link
Copy Markdown
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
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