Skip to content

Commit ccdc293

Browse files
committed
Fix type conversion for node_num in __get_mode_command method
1 parent 27406a8 commit ccdc293

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

superbench/runner/runner.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727

2828
class SuperBenchRunner():
2929
"""SuperBench runner class."""
30+
3031
def __init__(self, sb_config, docker_config, ansible_config, sb_output_dir):
3132
"""Initilize.
3233
@@ -155,7 +156,7 @@ def __get_mode_command(self, benchmark_name, mode, timeout=None):
155156
torch_dist_params = (
156157
f'--nnodes={mode.node_num} --rdzv-endpoint=$MASTER_ADDR:$MASTER_PORT '
157158
f'--rdzv-id={random.randint(100, 999)} --rdzv-backend=c10d '
158-
if 'node_num' in mode and mode.node_num > 1 else ''
159+
if 'node_num' in mode and int(mode.node_num) > 1 else ''
159160
)
160161

161162
nsys_prefix = (

0 commit comments

Comments
 (0)