From 1f898fcc584a37c89f73554aacbf2b6e0e3e7f87 Mon Sep 17 00:00:00 2001 From: yukirora Date: Wed, 7 Jul 2021 09:34:54 +0000 Subject: [PATCH] remove util related code --- tests/benchmarks/utils.py | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/tests/benchmarks/utils.py b/tests/benchmarks/utils.py index 6405f1ffe..b63d3cdbf 100644 --- a/tests/benchmarks/utils.py +++ b/tests/benchmarks/utils.py @@ -4,11 +4,12 @@ """Utilities for benchmark tests.""" import os +import socket +from contextlib import closing import multiprocessing as multiprocessing from multiprocessing import Process from superbench.benchmarks import BenchmarkRegistry -from superbench.common.utils import network def clean_simulated_ddp_distributed_env(): @@ -20,6 +21,18 @@ def clean_simulated_ddp_distributed_env(): os.environ.pop('MASTER_PORT') +def get_free_port(): + """Get a free port in current system. + + Return: + port (int): a free port in current system. + """ + with closing(socket.socket(socket.AF_INET, socket.SOCK_STREAM)) as s: + s.bind(('', 0)) + s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) + return s.getsockname()[1] + + def setup_simulated_ddp_distributed_env(world_size, local_rank, port): """Function to setup the simulated DDP distributed envionment variables.""" os.environ['WORLD_SIZE'] = str(world_size)