github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-06 16:53:01 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
e6943e2d11fbe7dd69c694e435f598cd140b1574
ColossalAI/tests/test_optimizer/test_fused_adam_kernel.py
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

2.6 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink