Skip to content

OP lowering for op native_dropout #5588

@zpcore

Description

@zpcore

Try manual lowing the op for native_dropout including adding the unit test.

As @JackCaoG has mentioned, native_dropout regress a lot model training on torchbench. We plan to add this into the 2.1 release.

Metadata

Metadata

Assignees

Labels

loweringATen Operation lowering

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions