github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-02-01 23:23:14 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
f0aa191f51704806e65ac849da137069bb35a6d5
ColossalAI/tests/test_optimizer
History
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
..
test_cpu_adam.py
[setup] support pre-build and jit-build of cuda kernels (#2374)
2023-01-06 20:50:26 +08:00
test_fused_adam_kernel.py
[setup] support pre-build and jit-build of cuda kernels (#2374)
2023-01-06 20:50:26 +08:00
test_fused_adam.py
[hotfix] fix CPUAdam kernel nullptr (#1410)
2022-08-05 19:45:45 +08:00
test_hybrid_adam.py
[hotfix] fix CPUAdam kernel nullptr (#1410)
2022-08-05 19:45:45 +08:00
test_nvme.py
[nvme] CPUAdam and HybridAdam support NVMe offload (#1360)
2022-07-26 17:25:24 +08:00