ColossalAI/extensions/csrc/funcs
傅剑寒 808ee6e4ad
[Inference/Feat] Feat quant kvcache step2 (#5674)
2024-04-30 11:26:36 +08:00
..
binary_functor.h [Inference/Refactor] Refactor compilation mechanism and unified multi hw (#5613) 2024-04-24 14:17:54 +08:00
cast_functor.h [Inference/Feat] Feat quant kvcache step2 (#5674) 2024-04-30 11:26:36 +08:00
reduce_function.h [Inference/Refactor] Refactor compilation mechanism and unified multi hw (#5613) 2024-04-24 14:17:54 +08:00
ternary_functor.h [Inference/Refactor] Refactor compilation mechanism and unified multi hw (#5613) 2024-04-24 14:17:54 +08:00
unary_functor.h [Inference/Feat] Add kvcache quantization support for FlashDecoding (#5656) 2024-04-26 19:40:37 +08:00