| 113 | } |
| 114 | |
| 115 | void PrintNodeDef(const NodeDef& node) { |
| 116 | std::cout << "===============" << std::endl; |
| 117 | std::cout << "name:id " << node.name_ << ":" << |
| 118 | node.id_ << std::endl; |
| 119 | std::cout << "pre: "; |
| 120 | for (int32_t pre : node.pre_) { |
| 121 | std::cout << pre << " "; |
| 122 | } |
| 123 | std::cout << std::endl; |
| 124 | std::cout << "succ: "; |
| 125 | for (int32_t succ : node.succ_) { |
| 126 | std::cout << succ << " "; |
| 127 | } |
| 128 | std::cout << std::endl; |
| 129 | std::cout << "input edges:" << std::endl; |
| 130 | for (EdgeDef input : node.input_edges_) { |
| 131 | std::cout << input.src_name_ << "," << |
| 132 | input.src_id_ << "," << |
| 133 | input.src_slot_ << std::endl; |
| 134 | } |
| 135 | if (node.name_ == "REMOTE") { |
| 136 | std::cout << "[" << std::endl; |
| 137 | for (NodeDef inner_node : |
| 138 | static_cast<const RemoteNodeDef*>(&node)->nodes_) { |
| 139 | PrintNodeDef(inner_node); |
| 140 | } |
| 141 | std::cout << "]" << std::endl; |
| 142 | } |
| 143 | } |
| 144 | |
| 145 | TEST(OptimizerTest, OptimizerShard) { |
| 146 | std::vector<std::vector<std::string>> fusion_output_map; |