1
0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-04-29 20:25:29 +00:00
ColossalAI/extensions/pybind/cpu_adam
傅剑寒 279300dc5f
[Inference/Refactor] Refactor compilation mechanism and unified multi hw ()
* refactor compilation mechanism and unified multi hw

* fix file path bug

* add init.py to make pybind a module to avoid relative path error caused by softlink

* delete duplicated micros

* fix micros bug in gcc
2024-04-24 14:17:54 +08:00
..
__init__.py [Inference/Refactor] Refactor compilation mechanism and unified multi hw () 2024-04-24 14:17:54 +08:00
cpu_adam_arm.py [Inference/Refactor] Refactor compilation mechanism and unified multi hw () 2024-04-24 14:17:54 +08:00
cpu_adam_x86.py [Inference/Refactor] Refactor compilation mechanism and unified multi hw () 2024-04-24 14:17:54 +08:00