mirror of https://github.com/hpcaitech/ColossalAI
40d376c566
* [setup] support pre-build and jit-build of cuda kernels * polish code * polish code * polish code * polish code * polish code * polish code |
||
---|---|---|
.. | ||
test_cpu_adam.py | ||
test_fused_adam.py | ||
test_fused_adam_kernel.py | ||
test_hybrid_adam.py | ||
test_nvme.py |