github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-20 12:43:55 +00:00
Code Issues Packages Projects Releases Wiki Activity
hotfix/kernel_build_before_load
ColossalAI/colossalai/kernel
History
wangbinluo 3942218618 remove useless platform args and comment
2024-01-11 08:21:53 +00:00
..
cuda_native [npu] use extension for op builder (#5172) 2024-01-08 11:39:16 +08:00
extensions fix the build before load bug 2024-01-10 14:50:40 +08:00
jit [npu] change device to accelerator api (#5239) 2024-01-09 10:20:05 +08:00
triton [Kernels]added flash-decoidng of triton (#5063) 2023-11-20 13:58:29 +08:00
__init__.py [npu] use extension for op builder (#5172) 2024-01-08 11:39:16 +08:00
base_kernel_loader.py [npu] use extension for op builder (#5172) 2024-01-08 11:39:16 +08:00
cpu_adam_loader.py remove useless platform args and comment 2024-01-11 08:21:53 +00:00
flash_attention_loader.py [npu] use extension for op builder (#5172) 2024-01-08 11:39:16 +08:00
op_builder [builder] reconfig op_builder for pypi install (#2314) 2023-01-04 16:32:32 +08:00