ColossalAI/extensions/csrc
傅剑寒 7ff42cc06d
add vec_type_trait implementation (#5473)
2024-03-19 18:36:40 +08:00
..
arm [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
common add vec_type_trait implementation (#5473) 2024-03-19 18:36:40 +08:00
cuda add vec_type_trait implementation (#5473) 2024-03-19 18:36:40 +08:00
x86 refactor code 2024-03-11 17:06:57 +08:00
__init__.py [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
scaled_softmax.py [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00