digger-yu
|
d96567bb5d
|
[misc] op_builder/builder.py (#3593)
Optimization Code
The source code has not been modified, only a few spelling errors in the comments have been changed
|
2 years ago |
Hongxin Liu
|
173dad0562
|
[misc] add verbose arg for zero and op builder (#3552)
* [misc] add print verbose
* [gemini] add print verbose
* [zero] add print verbose for low level
* [misc] add print verbose for op builder
|
2 years ago |
ver217
|
823f3b9cf4
|
[doc] add deepspeed citation and copyright (#2996)
* [doc] add deepspeed citation and copyright
* [doc] add deepspeed citation and copyright
* [doc] add deepspeed citation and copyright
|
2 years ago |
Frank Lee
|
3a5d93bc2c
|
[kernel] cached the op kernel and fixed version check (#2886)
* [kernel] cached the op kernel and fixed version check
* polish code
* polish code
|
2 years ago |
Frank Lee
|
dd14783f75
|
[kernel] fixed repeated loading of kernels (#2549)
* [kernel] fixed repeated loading of kernels
* polish code
* polish code
|
2 years ago |
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
|
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 |
Jiarui Fang
|
16cc8e6aa7
|
[builder] MOE builder (#2277)
|
2 years ago |
Jiarui Fang
|
54de05da5d
|
[builder] polish builder with better base class (#2216)
* [builder] polish builder
* remove print
|
2 years ago |
Jiarui Fang
|
7675792100
|
[builder] raise Error when CUDA_HOME is not set (#2213)
|
2 years ago |
Jiarui Fang
|
1cb532ffec
|
[builder] multihead attn runtime building (#2203)
* [hotfix] correcnt cpu_optim runtime compilation
* [builder] multihead attn
* fix bug
* fix a bug
|
2 years ago |
Jiarui Fang
|
d42afd30f8
|
[builder] runtime adam and fused_optim builder (#2184)
|
2 years ago |