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