apex / csrc

449 kB

Ctrl+K

1 contributor

History: 1 commit

camenduru

thanks to NVIDIA ❤

e828767 almost 3 years ago

megatron
thanks to NVIDIA ❤ almost 3 years ago
amp_C_frontend.cpp

5.93 kB
thanks to NVIDIA ❤ almost 3 years ago
compat.h

140 Bytes
thanks to NVIDIA ❤ almost 3 years ago
flatten_unflatten.cpp

584 Bytes
thanks to NVIDIA ❤ almost 3 years ago
fused_dense.cpp

7.98 kB
thanks to NVIDIA ❤ almost 3 years ago
fused_dense_cuda.cu

49.6 kB
thanks to NVIDIA ❤ almost 3 years ago
layer_norm_cuda.cpp

12.9 kB
thanks to NVIDIA ❤ almost 3 years ago
layer_norm_cuda_kernel.cu

35.9 kB
thanks to NVIDIA ❤ almost 3 years ago
mlp.cpp

4.74 kB
thanks to NVIDIA ❤ almost 3 years ago
mlp_cuda.cu

52.5 kB
thanks to NVIDIA ❤ almost 3 years ago
multi_tensor_adagrad.cu

3.04 kB
thanks to NVIDIA ❤ almost 3 years ago
multi_tensor_adam.cu

8.63 kB
thanks to NVIDIA ❤ almost 3 years ago
multi_tensor_apply.cuh

4.82 kB
thanks to NVIDIA ❤ almost 3 years ago
multi_tensor_axpby_kernel.cu

4.7 kB
thanks to NVIDIA ❤ almost 3 years ago
multi_tensor_l2norm_kernel.cu

12.8 kB
thanks to NVIDIA ❤ almost 3 years ago
multi_tensor_l2norm_kernel_mp.cu

5.95 kB
thanks to NVIDIA ❤ almost 3 years ago
multi_tensor_l2norm_scale_kernel.cu

9.26 kB
thanks to NVIDIA ❤ almost 3 years ago
multi_tensor_lamb.cu

12.7 kB
thanks to NVIDIA ❤ almost 3 years ago
multi_tensor_lamb_mp.cu

15.8 kB
thanks to NVIDIA ❤ almost 3 years ago
multi_tensor_lamb_stage_1.cu

4.4 kB
thanks to NVIDIA ❤ almost 3 years ago
multi_tensor_lamb_stage_2.cu

3.44 kB
thanks to NVIDIA ❤ almost 3 years ago
multi_tensor_novograd.cu

5.16 kB
thanks to NVIDIA ❤ almost 3 years ago
multi_tensor_scale_kernel.cu

4.17 kB
thanks to NVIDIA ❤ almost 3 years ago
multi_tensor_sgd_kernel.cu

8.12 kB
thanks to NVIDIA ❤ almost 3 years ago
syncbn.cpp

6.25 kB
thanks to NVIDIA ❤ almost 3 years ago
type_shim.h

10.1 kB
thanks to NVIDIA ❤ almost 3 years ago
welford.cu

54.6 kB
thanks to NVIDIA ❤ almost 3 years ago