7 lines
309 B
Markdown
7 lines
309 B
Markdown
|
This CUDA extensions implements fused dropout + residual + LayerNorm, based on
|
||
|
|
Apex's [FastLayerNorm](https://github.com/NVIDIA/apex/tree/master/apex/contrib/layer_norm).
|
||
|
|
We add dropout and residual, and make it work for both pre-norm and post-norm architecture.
|
||
|
|
```sh
|
||
|
|
cd csrc/layer_norm && pip install .
|
||
|
|
```
|