ColossalAI/colossalai/fx/tracer/bias_addition_patch
YuliangLiu0306 e34e850a4c
[autoparallel]add essential CommActions for broadcast oprands (#1793)
2022-11-04 18:36:42 +08:00
..
patched_bias_addition_function [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00
patched_bias_addition_module [autoparallel]add essential CommActions for broadcast oprands (#1793) 2022-11-04 18:36:42 +08:00
__init__.py [fx] support module with bias addition (#1780) 2022-11-01 22:53:51 +08:00