Skip to content

Enhance MemoryEfficientSwish with optional import of SiLU #2812

@yiheng-wang-nv

Description

@yiheng-wang-nv

From Pytorch 1.7, SiLU is implemented, it is the optimized version of Swish.

In addition Mish needs to support the parameter inplace when using torch's native version.

Metadata

Metadata

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions