()
| 102 | |
| 103 | |
| 104 | def node_size(): |
| 105 | node_num_from_env = os.getenv("ONEFLOW_TEST_NODE_NUM", None) |
| 106 | if node_num_from_env: |
| 107 | return int(node_num_from_env) |
| 108 | elif has_node_list(): |
| 109 | node_list_from_env = node_list() |
| 110 | return len(node_list_from_env) |
| 111 | else: |
| 112 | return 1 |
| 113 | |
| 114 | |
| 115 | def has_world_size(): |
no test coverage detected