github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-08 19:38:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
hotfix/kernel_build_before_load
ColossalAI/colossalai/kernel/extensions
History
wangbinluo a9b5ec8664 fix the build before load bug
2024-01-10 14:50:40 +08:00
..
cpu_adam fix the build before load bug 2024-01-10 14:50:40 +08:00
flash_attention [npu] change device to accelerator api (#5239) 2024-01-09 10:20:05 +08:00
__init__.py [npu] use extension for op builder (#5172) 2024-01-08 11:39:16 +08:00
base_extension.py [npu] use extension for op builder (#5172) 2024-01-08 11:39:16 +08:00
extension_builder.py [npu] use extension for op builder (#5172) 2024-01-08 11:39:16 +08:00
utils.py [npu] use extension for op builder (#5172) 2024-01-08 11:39:16 +08:00