ColossalAI/colossalai/utils/multi_tensor_apply at 116e3d0b8f77249217d4a36cecf629e1e688b84e - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-18 00:17:58 +00:00

Files

History

Jiarui Fang 355ffb386e [builder] unified cpu_optim fused_optim inferface (#2190 )

2022-12-23 20:57:41 +08:00

..

__init__.py

Develop/experiments (#59 )

2021-12-09 15:08:29 +08:00

multi_tensor_apply.py

[builder] unified cpu_optim fused_optim inferface (#2190 )

2022-12-23 20:57:41 +08:00