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