githubmirror/ColossalAI | Gitverse

/

/

Обзор ДокументацияВойти

/

/

Аналитика

Безопасность

tests/test_optimizer/

test_cpu_adam.py

[setup] support pre-build and jit-build of cuda kernels (#2374)

4 года назад

test_fused_adam.py

[hotfix] fix CPUAdam kernel nullptr (#1410)

4 года назад

test_fused_adam_kernel.py

[setup] support pre-build and jit-build of cuda kernels (#2374)

4 года назад

test_hybrid_adam.py

[hotfix] fix CPUAdam kernel nullptr (#1410)

4 года назад

[nvme] CPUAdam and HybridAdam support NVMe offload (#1360)

4 года назад