ColossalAI/extensions/csrc at ed5ebd1735db4541709eebdd37839ad161f542e8 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-13 03:38:19 +00:00

Files

History

傅剑寒 7ebdf48ac5 add cast and op_functor for cuda build-in types (#5546 )

2024-04-08 11:38:05 +08:00

..

[feat] refactored extension module (#5298 )

2024-01-25 17:01:48 +08:00

[Inference] Add Reduce Utils (#5537 )

2024-04-01 15:34:25 +08:00

add cast and op_functor for cuda build-in types (#5546 )

2024-04-08 11:38:05 +08:00

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