()
| 272 | |
| 273 | |
| 274 | def server(): |
| 275 | print(f"[ server ] starting ...", flush=True) |
| 276 | parser = argparse.ArgumentParser() |
| 277 | parser.add_argument( |
| 278 | "--channel-ip", |
| 279 | type=str, |
| 280 | default=None, |
| 281 | help="IP for ZeroMQ channel", |
| 282 | ) |
| 283 | parser.add_argument( |
| 284 | "--channel-port", |
| 285 | type=int, |
| 286 | default=5555, |
| 287 | help="Port for ZeroMQ channel", |
| 288 | ) |
| 289 | parser.add_argument( |
| 290 | "--master-port", |
| 291 | type=int, |
| 292 | default=6666, |
| 293 | help="Port for torch distributed", |
| 294 | ) |
| 295 | parser.add_argument( |
| 296 | "--samples-per-problem", |
| 297 | type=int, |
| 298 | default=200, |
| 299 | help="Number of samples to generate for each problem", |
| 300 | ) |
| 301 | parser.add_argument( |
| 302 | "--gen-node-world-size", |
| 303 | type=int, |
| 304 | default=1, |
| 305 | help="Number of machines to use for generation", |
| 306 | ) |
| 307 | parser.add_argument( |
| 308 | "--src-path", |
| 309 | type=str, |
| 310 | default="", |
| 311 | help="Get source path", |
| 312 | ) |
| 313 | parser.add_argument( |
| 314 | "--tgt-path", |
| 315 | type=str, |
| 316 | default="", |
| 317 | help="Get target path", |
| 318 | ) |
| 319 | parser.add_argument( |
| 320 | "--problem-split", |
| 321 | type=str, |
| 322 | default="test", |
| 323 | ) |
| 324 | parser.add_argument( |
| 325 | "--micro-batch-size", |
| 326 | type=int, |
| 327 | default=1, |
| 328 | ) |
| 329 | parser.add_argument( |
| 330 | '--language-src-type', |
| 331 | type=str, |
nothing calls this directly
no test coverage detected