github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-05 02:51:59 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
v0.2.1
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