|
|
@@ -50,6 +50,7 @@ class Cluster { |
|
|
|
~Cluster() = default; |
|
|
|
}; |
|
|
|
using ClusterPtr = std::shared_ptr<Cluster>; |
|
|
|
using PartitionModeMap = std::map<Mode, std::string>; |
|
|
|
|
|
|
|
class GraphPartitioner { |
|
|
|
public: |
|
|
@@ -176,6 +177,8 @@ class GraphPartitioner { |
|
|
|
Graph2InputNodesSubGraphInfo graph_2_input_subgraph_; |
|
|
|
GraphPartitionInfo graph_info_; |
|
|
|
uint32_t partition_times_; // times of call partition |
|
|
|
PartitionModeMap mode_2_string_ = {{kPartitioning, "Partitioning"}, |
|
|
|
{kSecondPartitioning, "SecondPartitioning"},{kMerging, "Merging"}}; |
|
|
|
friend class GraphManager; |
|
|
|
}; |
|
|
|
} // namespace ge |
|
|
|