You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/kernel
Xu Kai 785802e809
[inference] add reference and fix some bugs (#4937)
1 year ago
..
cuda_native [kernel] support pure fp16 for cpu adam and update gemini optim tests (#4921) 1 year ago
jit [misc] update pre-commit and run all files (#4752) 1 year ago
triton [inference] add reference and fix some bugs (#4937) 1 year ago
__init__.py [hotfix] Fix import error: colossal.kernel without triton installed (#4722) 1 year ago
op_builder [builder] reconfig op_builder for pypi install (#2314) 2 years ago