github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/fx/tracer/meta_patch/patched_function
History
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
 		2022-11-01 22:53:51 +08:00
..
__init__.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
activation_function.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
arithmetic.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
convolution.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
embedding.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
normalization.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
python_ops.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
torch_ops.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00