in bench_cluster/network_bench.py [0:0]
def network_bench(
out_dir: str,
gpus: int,
qos: str,
trials: int,
warmups: int,
maxsize: int,
async_op: bool,
bw_unit: str,
scan: bool,
raw: bool,
dtype: str,
mem_factor: float,
debug: bool = False,