mirror of https://github.com/hpcaitech/ColossalAI
![]() * [autoparallel] support distributed dataloader option * update output handler to support ddp dataloader * poish code |
||
---|---|---|
.. | ||
__init__.py | ||
batch_norm_generator.py | ||
binary_elementwise_generator.py | ||
conv_strategy_generator.py | ||
getattr_generator.py | ||
getitem_generator.py | ||
layer_norm_generator.py | ||
matmul_strategy_generator.py | ||
normal_pooling_generator.py | ||
output_generator.py | ||
placeholder_generator.py | ||
reshape_generator.py | ||
strategy_generator.py | ||
unary_elementwise_generator.py | ||
where_generator.py |