Skip to content

Commit 334e9a4

Browse files
committed
Empty lines
1 parent 54b4de4 commit 334e9a4

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,7 +27,6 @@
2727

2828
class SuperBenchRunner():
2929
"""SuperBench runner class."""
30-
3130
def __init__(self, sb_config, docker_config, ansible_config, sb_output_dir):
3231
"""Initilize.
3332
@@ -158,10 +157,12 @@ def __get_mode_command(self, benchmark_name, mode, timeout=None):
158157
f'--rdzv-id={random.randint(100, 999)} --rdzv-backend=c10d '
159158
if 'node_num' in mode and mode.node_num > 1 else ''
160159
)
160+
161161
nsys_prefix = (
162162
f'nsys profile --output {trace_dir}/{benchmark_name}_traces '
163163
f'--backtrace none --sample none --force-overwrite true --cpuctxsw none --trace cuda,nvtx '
164164
) if enable_nsys else ''
165+
165166
mode_command = (
166167
f'{nsys_prefix}'
167168
f'torchrun'

0 commit comments

Comments
 (0)