github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-02 09:38:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

[autoparallel] adapt autoparallel tests with latest api (#2626)

Browse Source
This commit is contained in:
YuliangLiu0306
2023-02-08 15:02:12 +08:00
committed by GitHub
parent c375563653
commit cb3d1bef62
8 changed files with 59 additions and 583 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
colossalai/auto_parallel/tensor_shard/solver/cost_graph.py
View File

@@ -62,9 +62,6 @@ class CostGraph:
else:
edge_cost[(j, i)] = resharding_cost_item.total
self.edge_costs[node_pair] = edge_cost
# add parents and children attribute to node
# parent_nodes = [node for node in strategies_vector.predecessor_nodes]
# children_nodes = [node for node in strategies_vector.successor_nodes]
parent_nodes = []
children_nodes = []