Hongxin Liu
|
079bf3cb26
|
[misc] update pre-commit and run all files (#4752)
* [misc] update pre-commit
* [misc] run pre-commit
* [misc] remove useless configuration files
* [misc] ignore cuda for clang-format
|
1 year ago |
YuliangLiu0306
|
fea3cb661c
|
[autoparallel] support addmm in tracer and solver (#1961)
* [fx] patch addmm
* [autoparallel] support addmm in tracer and solver
|
2 years ago |
YuliangLiu0306
|
f6032ddb17
|
[autoparallel] fix bias addition module (#1800)
|
2 years ago |
YuliangLiu0306
|
e859380bf7
|
[fx] support module with bias addition (#1780)
* [autoparallel] refactor tracer to fix bias addition issue
* [fx] support module with bias addition
* create bias_addition_module
* refactor file structure
* polish code
* fix unit test
|
2 years ago |
Frank Lee
|
262652c8bc
|
[autoparallel] added addbmm handler (#1751)
|
2 years ago |
Xue Fuzhao
|
e070ca45c6
|
[NFC] polish colossalai/fx/tracer/meta_patch/patched_module/convolution.py code style (#1563)
|
2 years ago |
YuliangLiu0306
|
4b3d6caeb3
|
[fx]patch nn.functional convolution (#1528)
|
2 years ago |
YuliangLiu0306
|
413c053453
|
[autoparallel] add cost graph class (#1481)
* [autoparallel] add cost graph class
* polish code
|
2 years ago |
Frank Lee
|
3da68d6b1b
|
[fx] fixed adapative pooling size concatenation error (#1489)
|
2 years ago |
Frank Lee
|
7d6293927f
|
[fx] patched torch.max and data movement operator (#1391)
* [fx] patched torch.max and data movement operator
* polish code
|
2 years ago |
Frank Lee
|
ad678921db
|
[fx] patched torch.full for huggingface opt (#1386)
|
2 years ago |
Super Daniel
|
be229217ce
|
[fx] add torchaudio test (#1369)
* [fx]add torchaudio test
* [fx]add torchaudio test
* [fx] add torchaudio test
* [fx] add torchaudio test
* [fx] add torchaudio test
* [fx] add torchaudio test
* [fx] add torchaudio test
* [fx] add torchaudio test and test patches
* Delete ~
* [fx] add patches and patches test
* [fx] add patches and patches test
* [fx] fix patches
* [fx] fix rnn patches
* [fx] fix rnn patches
* [fx] fix rnn patches
* [fx] fix rnn patches
* [fx] merge upstream
* [fx] fix import errors
|
2 years ago |
Frank Lee
|
274c1a3b5f
|
[fx] fixed apex normalization patch exception (#1352)
|
2 years ago |
YuliangLiu0306
|
942c8cd1fb
|
[fx] refactor tracer to trace complete graph (#1342)
* [CLI] add CLI launcher
* Revert "[CLI] add CLI launcher"
This reverts commit df7e6506d4 .
* [fx] refactor tracer to trace complete graph
* add comments and solve conflicts.
|
2 years ago |
YuliangLiu0306
|
4631fef8a0
|
[fx]refactor tracer (#1335)
|
2 years ago |
Frank Lee
|
75abc75c15
|
[fx] fixed compatiblity issue with torch 1.10 (#1331)
|
2 years ago |
Frank Lee
|
4f4d8c3656
|
[fx] added apex normalization to patched modules (#1300)
* [fx] added apex normalization to patched modules
* remove unused imports
|
2 years ago |
Frank Lee
|
7531c6271f
|
[fx] refactored the file structure of patched function and module (#1238)
* [fx] refactored the file structure of patched function and module
* polish code
|
2 years ago |
Frank Lee
|
84f2298a96
|
[fx] added patches for tracing swin transformer (#1228)
|
2 years ago |
Frank Lee
|
b6cb5a47ad
|
[fx] added timm model tracing testing (#1221)
|
2 years ago |
Frank Lee
|
11973d892d
|
[fx] added torchvision model tracing testing (#1216)
* [fx] added torchvision model tracing testing
* remove unused imports
|
2 years ago |
Frank Lee
|
5da87ce35d
|
[fx] added testing for all albert variants (#1211)
|
2 years ago |
Frank Lee
|
2d13a45a3b
|
[fx] added testing for all gpt variants (#1210)
* [fx] added testing for all gpt variants
* polish code
* polish code
|
2 years ago |
Frank Lee
|
f7878f465c
|
[fx] supported model tracing for huggingface bert (#1201)
* [fx] supported model tracing for huggingface bert
* polish test
|
2 years ago |
Frank Lee
|
abf6a262dc
|
[fx] added module patch for pooling layers (#1197)
|
2 years ago |
Frank Lee
|
2c8c05675d
|
[fx] patched conv and normalization (#1188)
|
2 years ago |
Frank Lee
|
6d86f1bc91
|
[fx] supported data-dependent control flow in model tracing (#1185)
* [fx] supported data-dependent control flow in model tracing
* polish code
|
2 years ago |