camenduru
/

apex

Model card Files Files and versions

apex / csrc /megatron

105 kB

1 contributor

History: 1 commit

camenduru's picture

thanks to NVIDIA ❤

e828767 almost 3 years ago

fused_weight_gradient_dense.cpp

540 Bytes

thanks to NVIDIA ❤ almost 3 years ago
fused_weight_gradient_dense_16bit_prec_cuda.cu

3.52 kB

thanks to NVIDIA ❤ almost 3 years ago
fused_weight_gradient_dense_cuda.cu

4.28 kB

thanks to NVIDIA ❤ almost 3 years ago
generic_scaled_masked_softmax.cpp

2.95 kB

thanks to NVIDIA ❤ almost 3 years ago
generic_scaled_masked_softmax.h

12.6 kB

thanks to NVIDIA ❤ almost 3 years ago
generic_scaled_masked_softmax_cuda.cu

3.83 kB

thanks to NVIDIA ❤ almost 3 years ago
scaled_masked_softmax.cpp

3.37 kB

thanks to NVIDIA ❤ almost 3 years ago
scaled_masked_softmax.h

32.7 kB

thanks to NVIDIA ❤ almost 3 years ago
scaled_masked_softmax_cuda.cu

4.21 kB

thanks to NVIDIA ❤ almost 3 years ago
scaled_softmax.cpp

2.47 kB

thanks to NVIDIA ❤ almost 3 years ago
scaled_softmax_cuda.cu

3.28 kB

thanks to NVIDIA ❤ almost 3 years ago
scaled_upper_triang_masked_softmax.cpp

2.55 kB

thanks to NVIDIA ❤ almost 3 years ago
scaled_upper_triang_masked_softmax.h

25.3 kB

thanks to NVIDIA ❤ almost 3 years ago
scaled_upper_triang_masked_softmax_cuda.cu

3.16 kB

thanks to NVIDIA ❤ almost 3 years ago