ColossalAI/colossalai/auto_parallel/solver/strategy
Frank Lee 30e50c8b4a
[autoparallel] implemented all matmul strategy generator (#1650)
2022-09-27 12:06:25 +08:00
..
__init__.py [autoparallel] implemented linear projection strategy generator (#1639) 2022-09-26 16:58:14 +08:00
matmul_strategy_generator.py [autoparallel] implemented all matmul strategy generator (#1650) 2022-09-27 12:06:25 +08:00
strategy_generator.py [autoparallel] implemented all matmul strategy generator (#1650) 2022-09-27 12:06:25 +08:00