github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-03 01:55:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

[autoparallel]add backward cost info into strategies (#1524)

Browse Source
This commit is contained in:
YuliangLiu0306
2022-09-07 11:19:00 +08:00
committed by GitHub
parent 1a3599410d
commit 0908d0fc61
3 changed files with 142 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_auto_parallel/test_conv_handler.py
View File

@@ -82,7 +82,6 @@ def test_conv_handler():
strategies_vector=strategies_vector,
shape_consistency_manager=shape_consistency_manager)
conv_handler.register_strategy()
# ['S0S1 = S0R x RS1', 'S1S0 = S1R x RS0', 'S0R = S0R x RR', 'S1R = S1R x RR', 'S0R = S0S1 x S1R', 'S1R = S1S0 x S0R', 'RS1 = RS0 x S0S1', 'RS0 = RS1 x S1S0', 'RR = RS0 x S0R', 'RR = RS1 x S1R', 'RS0 = RR x RS0', 'RS1 = RR x RS1', 'RR = RR x RR', 'S01R = S01R x RR', 'RR = RS01 x S01R']
strategy_name_list = [strategy.name for strategy in conv_handler.strategies_vector]