apex / csrc

449 kB

1 contributor

History: 1 commit

camenduru

thanks to NVIDIA ❤

e828767 over 2 years ago

megatron
thanks to NVIDIA ❤ over 2 years ago
amp_C_frontend.cpp

5.93 kB

thanks to NVIDIA ❤ over 2 years ago
compat.h

140 Bytes

thanks to NVIDIA ❤ over 2 years ago
flatten_unflatten.cpp

584 Bytes

thanks to NVIDIA ❤ over 2 years ago
fused_dense.cpp

7.98 kB

thanks to NVIDIA ❤ over 2 years ago
fused_dense_cuda.cu

49.6 kB

thanks to NVIDIA ❤ over 2 years ago
layer_norm_cuda.cpp

12.9 kB

thanks to NVIDIA ❤ over 2 years ago
layer_norm_cuda_kernel.cu

35.9 kB

thanks to NVIDIA ❤ over 2 years ago
mlp.cpp

4.74 kB

thanks to NVIDIA ❤ over 2 years ago
mlp_cuda.cu

52.5 kB

thanks to NVIDIA ❤ over 2 years ago
multi_tensor_adagrad.cu

3.04 kB

thanks to NVIDIA ❤ over 2 years ago
multi_tensor_adam.cu

8.63 kB

thanks to NVIDIA ❤ over 2 years ago
multi_tensor_apply.cuh

4.82 kB

thanks to NVIDIA ❤ over 2 years ago
multi_tensor_axpby_kernel.cu

4.7 kB

thanks to NVIDIA ❤ over 2 years ago
multi_tensor_l2norm_kernel.cu

12.8 kB

thanks to NVIDIA ❤ over 2 years ago
multi_tensor_l2norm_kernel_mp.cu

5.95 kB

thanks to NVIDIA ❤ over 2 years ago
multi_tensor_l2norm_scale_kernel.cu

9.26 kB

thanks to NVIDIA ❤ over 2 years ago
multi_tensor_lamb.cu

12.7 kB

thanks to NVIDIA ❤ over 2 years ago
multi_tensor_lamb_mp.cu

15.8 kB

thanks to NVIDIA ❤ over 2 years ago
multi_tensor_lamb_stage_1.cu

4.4 kB

thanks to NVIDIA ❤ over 2 years ago
multi_tensor_lamb_stage_2.cu

3.44 kB

thanks to NVIDIA ❤ over 2 years ago
multi_tensor_novograd.cu

5.16 kB

thanks to NVIDIA ❤ over 2 years ago
multi_tensor_scale_kernel.cu

4.17 kB

thanks to NVIDIA ❤ over 2 years ago
multi_tensor_sgd_kernel.cu

8.12 kB

thanks to NVIDIA ❤ over 2 years ago
syncbn.cpp

6.25 kB

thanks to NVIDIA ❤ over 2 years ago
type_shim.h

10.1 kB

thanks to NVIDIA ❤ over 2 years ago
welford.cu

54.6 kB

thanks to NVIDIA ❤ over 2 years ago