csrc/megatron/scaled_upper_triang_masked_softmax_cuda.cu · f79993d92cfefbaf20a670dcbd61749ac70aed28 · OpenDAS / apex

Masaki Kozuki authored Oct 02, 2021


Co-authored-by: Piotr Bialecki <pbialecki@nvidia.com>
Co-authored-by: Eddie Yan <eddiey@nvidia.com>
Co-authored-by: Rishi Puri <riship@nvidia.com>
Co-authored-by: Sangkug Lym <slym@nvidia.com>

365fdc18

scaled_upper_triang_masked_softmax_cuda.cu 3.08 KB

Replace scaled_upper_triang_masked_softmax_cuda.cu