github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-01 21:26:42 +00:00
Code Issues Packages Projects Releases Wiki Activity
main
ColossalAI/extensions/pybind/layernorm
History
傅剑寒 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
layer_norm.cpp [Inference/Refactor] Refactor compilation mechanism and unified multi hw () 2024-04-24 14:17:54 +08:00
layernorm_cuda.py [Inference/Refactor] Refactor compilation mechanism and unified multi hw () 2024-04-24 14:17:54 +08:00