|
|
@@ -57,6 +57,8 @@ class GraphPartitioner { |
|
|
|
/// MergeAfterSubGraphOptimization() can only be called in Merge mode. |
|
|
|
/// After Partition(), change to Merge mode. After MergeAfterSubGraphOptimization(), change to Partition mode |
|
|
|
enum Mode { kPartitioning, kSecondPartitioning, kMerging }; |
|
|
|
std::map<Mode, std::string> mode2str = |
|
|
|
{{kPartitioning, "kPartitioning"}, {kSecondPartitioning, "kSecondPartitioning"}, {kMerging, "kMerging"}}; |
|
|
|
GraphPartitioner() : partition_times_(0){}; |
|
|
|
~GraphPartitioner() = default; |
|
|
|
|
|
|
|