zhengbochao 6c7f7663e1 Initial commit il y a 2 semaines
..
megatron 6c7f7663e1 Initial commit il y a 2 semaines
amp_C_frontend.cpp 6c7f7663e1 Initial commit il y a 2 semaines
compat.h 6c7f7663e1 Initial commit il y a 2 semaines
flatten_unflatten.cpp 6c7f7663e1 Initial commit il y a 2 semaines
fused_dense.cpp 6c7f7663e1 Initial commit il y a 2 semaines
fused_dense_cuda.cu 6c7f7663e1 Initial commit il y a 2 semaines
layer_norm_cuda.cpp 6c7f7663e1 Initial commit il y a 2 semaines
layer_norm_cuda_kernel.cu 6c7f7663e1 Initial commit il y a 2 semaines
mlp.cpp 6c7f7663e1 Initial commit il y a 2 semaines
mlp_cuda.cu 6c7f7663e1 Initial commit il y a 2 semaines
multi_tensor_adagrad.cu 6c7f7663e1 Initial commit il y a 2 semaines
multi_tensor_adam.cu 6c7f7663e1 Initial commit il y a 2 semaines
multi_tensor_apply.cuh 6c7f7663e1 Initial commit il y a 2 semaines
multi_tensor_axpby_kernel.cu 6c7f7663e1 Initial commit il y a 2 semaines
multi_tensor_l2norm_kernel.cu 6c7f7663e1 Initial commit il y a 2 semaines
multi_tensor_l2norm_kernel_mp.cu 6c7f7663e1 Initial commit il y a 2 semaines
multi_tensor_l2norm_scale_kernel.cu 6c7f7663e1 Initial commit il y a 2 semaines
multi_tensor_lamb.cu 6c7f7663e1 Initial commit il y a 2 semaines
multi_tensor_lamb_mp.cu 6c7f7663e1 Initial commit il y a 2 semaines
multi_tensor_lamb_stage_1.cu 6c7f7663e1 Initial commit il y a 2 semaines
multi_tensor_lamb_stage_2.cu 6c7f7663e1 Initial commit il y a 2 semaines
multi_tensor_novograd.cu 6c7f7663e1 Initial commit il y a 2 semaines
multi_tensor_scale_kernel.cu 6c7f7663e1 Initial commit il y a 2 semaines
multi_tensor_sgd_kernel.cu 6c7f7663e1 Initial commit il y a 2 semaines
static_switch.h 6c7f7663e1 Initial commit il y a 2 semaines
syncbn.cpp 6c7f7663e1 Initial commit il y a 2 semaines
type_shim.h 6c7f7663e1 Initial commit il y a 2 semaines
update_scale_hysteresis.cu 6c7f7663e1 Initial commit il y a 2 semaines
welford.cu 6c7f7663e1 Initial commit il y a 2 semaines