.. |
kernels
|
[misc] update pre-commit and run all files (#4752)
|
2023-09-19 14:20:26 +08:00 |
colossal_C_frontend.cpp
|
…
|
|
compat.h
|
[misc] update pre-commit and run all files (#4752)
|
2023-09-19 14:20:26 +08:00 |
cpu_adam.cpp
|
…
|
|
cpu_adam.h
|
…
|
|
layer_norm_cuda.cpp
|
[misc] update pre-commit and run all files (#4752)
|
2023-09-19 14:20:26 +08:00 |
layer_norm_cuda_kernel.cu
|
[misc] update pre-commit and run all files (#4752)
|
2023-09-19 14:20:26 +08:00 |
moe_cuda.cpp
|
[misc] update pre-commit and run all files (#4752)
|
2023-09-19 14:20:26 +08:00 |
moe_cuda_kernel.cu
|
[misc] update pre-commit and run all files (#4752)
|
2023-09-19 14:20:26 +08:00 |
multi_tensor_adam.cu
|
[doc] add deepspeed citation and copyright (#2996)
|
2023-03-04 20:08:11 +08:00 |
multi_tensor_apply.cuh
|
[doc] add deepspeed citation and copyright (#2996)
|
2023-03-04 20:08:11 +08:00 |
multi_tensor_l2norm_kernel.cu
|
[misc] update pre-commit and run all files (#4752)
|
2023-09-19 14:20:26 +08:00 |
multi_tensor_lamb.cu
|
[misc] update pre-commit and run all files (#4752)
|
2023-09-19 14:20:26 +08:00 |
multi_tensor_scale_kernel.cu
|
[misc] update pre-commit and run all files (#4752)
|
2023-09-19 14:20:26 +08:00 |
multi_tensor_sgd_kernel.cu
|
[misc] update pre-commit and run all files (#4752)
|
2023-09-19 14:20:26 +08:00 |
multihead_attention_1d.cpp
|
…
|
|
multihead_attention_1d.h
|
…
|
|
scaled_masked_softmax.cpp
|
[misc] update pre-commit and run all files (#4752)
|
2023-09-19 14:20:26 +08:00 |
scaled_masked_softmax.h
|
[misc] update pre-commit and run all files (#4752)
|
2023-09-19 14:20:26 +08:00 |
scaled_masked_softmax_cuda.cu
|
…
|
|
scaled_upper_triang_masked_softmax.cpp
|
…
|
|
scaled_upper_triang_masked_softmax.h
|
[misc] update pre-commit and run all files (#4752)
|
2023-09-19 14:20:26 +08:00 |
scaled_upper_triang_masked_softmax_cuda.cu
|
…
|
|
type_shim.h
|
[bf16] add bf16 support (#3882)
|
2023-06-05 15:58:31 +08:00 |