3 Commits (7e4de520e16af2f555fee760f158ef9e55d80b12)

Author SHA1 Message Date
Frank Lee 40d376c566
[setup] support pre-build and jit-build of cuda kernels (#2374) 2 years ago
Jiarui Fang db6eea3583
[builder] reconfig op_builder for pypi install (#2314) 2 years ago
Frank Lee 9b765e7a69
[setup] removed the build dependency on colossalai (#2307) 2 years ago
Jiarui Fang 54de05da5d
[builder] polish builder with better base class (#2216) 2 years ago
Jiarui Fang 7675792100
[builder] raise Error when CUDA_HOME is not set (#2213) 2 years ago
Jiarui Fang 5682e6d346
[hotfix] correcnt cpu_optim runtime compilation (#2197) 2 years ago
Jiarui Fang bc0e271e71
[buider] use builder() for cpu adam and fused optim in setup.py (#2187) 2 years ago
Jiarui Fang d42afd30f8
[builder] runtime adam and fused_optim builder (#2184) 2 years ago