ColossalAI/kernel at hotfix/kernel_build_before_load - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-26 12:14:02 +00:00

Files

History

wangbinluo 3942218618 remove useless platform args and comment

2024-01-11 08:21:53 +00:00

..

[npu] use extension for op builder (#5172 )

2024-01-08 11:39:16 +08:00

fix the build before load bug

2024-01-10 14:50:40 +08:00

[npu] change device to accelerator api (#5239 )

2024-01-09 10:20:05 +08:00

[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