ColossalAI/extensions/csrc/cuda/include
xs_courtesy 95c21498d4 add silu_and_mul for infer 2024-03-07 16:57:49 +08:00
..
block_reduce.h [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
mp_type_traits.h add silu_and_mul for infer 2024-03-07 16:57:49 +08:00