csrc/layer_norm_cuda_kernel.cu · 27a473459c96943d4c046ad53f413f85570a8955 · OpenDAS / apex

"example/37_permute/common.hpp" did not exist on "19147f593e095b9eb9a27fc3b5894f664d1bdccf"

Hubert Lu authored Jan 25, 2022

* Optimize fused layer normalization for MI100

* Optimize cuComputePartGradGammaBeta for AMD GPUs

1cb3da87

layer_norm_cuda_kernel.cu 27.6 KB