github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/extensions/csrc/cuda
History
xs_courtesy 01d289d8e5 Merge branch 'feature/colossal-infer' of https://github.com/hpcaitech/ColossalAI into add_gpu_launch_config 2024-03-08 15:04:55 +08:00
..
include add silu_and_mul for infer 2024-03-07 16:57:49 +08:00
utils add reusable utils for cuda 2024-03-08 14:53:29 +08:00
activation_kernel.cu add silu_and_mul for infer 2024-03-07 16:57:49 +08:00
colossal_C_frontend.cpp [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
colossal_inference_C_frontend.cpp add silu_and_mul for infer 2024-03-07 16:57:49 +08:00
compat.h [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
cpu_adam.cpp [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
cpu_adam.h [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
decode_kv_cache_memcpy_kernel.cu [Inference]Add CUDA KVCache Kernel (#5406) 2024-02-28 14:36:50 +08:00
layer_norm_cuda.cpp [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
layer_norm_cuda_kernel.cu [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
moe_cuda.cpp [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
moe_cuda_kernel.cu [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
multi_tensor_adam.cu [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
multi_tensor_apply.cuh [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
multi_tensor_l2norm_kernel.cu [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
multi_tensor_lamb.cu [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
multi_tensor_scale_kernel.cu [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
multi_tensor_sgd_kernel.cu [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
scaled_masked_softmax.cpp [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
scaled_masked_softmax.h [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
scaled_masked_softmax_cuda.cu [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
scaled_upper_triang_masked_softmax.cpp [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
scaled_upper_triang_masked_softmax.h [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
scaled_upper_triang_masked_softmax_cuda.cu [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
type_shim.h add silu_and_mul for infer 2024-03-07 16:57:49 +08:00