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