ColossalAI/colossalai/nn/layer/moe
Frank Lee 40d376c566
[setup] support pre-build and jit-build of cuda kernels (#2374)
* [setup] support pre-build and jit-build of cuda kernels

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code
2023-01-06 20:50:26 +08:00
..
__init__.py [moe] fix moe bugs (#1633) 2022-09-23 15:33:57 +08:00
_operation.py [setup] support pre-build and jit-build of cuda kernels (#2374) 2023-01-06 20:50:26 +08:00
experts.py [moe] fix MoE bugs (#1628) 2022-09-22 13:56:30 +08:00
layers.py [moe] fix moe bugs (#1633) 2022-09-23 15:33:57 +08:00
routers.py [moe] fix moe bugs (#1633) 2022-09-23 15:33:57 +08:00
utils.py [gemini] add GeminiMemoryManger (#832) 2022-04-24 13:08:48 +08:00