Commit Graph

9 Commits (b4788d63ed3bc8e650c878df24f86c8ddaed124e)

Author SHA1 Message Date
digger-yu d96567bb5d
[misc] op_builder/builder.py (#3593)
2 years ago
Hongxin Liu 173dad0562
[misc] add verbose arg for zero and op builder (#3552)
2 years ago
ver217 823f3b9cf4
[doc] add deepspeed citation and copyright (#2996)
2 years ago
Frank Lee 3a5d93bc2c
[kernel] cached the op kernel and fixed version check (#2886)
2 years ago
Frank Lee dd14783f75
[kernel] fixed repeated loading of kernels (#2549)
2 years ago
Frank Lee 40d376c566
[setup] support pre-build and jit-build of cuda kernels (#2374)
2 years ago
Frank Lee 8711310cda
[setup] remove torch dependency (#2333)
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