csrc/multi_tensor_l2norm_scale_kernel.cu · cc92a4b47dc45a6badb384ce2c68e43940e380fa · OpenDAS / apex

Burc Eryilmaz authored Sep 01, 2021



* fuse norm into scale

* add fused norm into dlamb
Co-authored-by: Sukru Eryilmaz <seryilmaz@computelab-dgx1v-32.nvidia.com>

4d190db6

multi_tensor_l2norm_scale_kernel.cu 9.04 KB