Skip to content

changing the command to run TB schmoo to get better numbers #232

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
urtiwari merged 2 commits into from
Jun 19, 2026
Merged

changing the command to run TB schmoo to get better numbers #232

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
a699e5e
changing the command to run TB schmoo to get better numbers
Jun 19, 2026
19af19d
changing the regex to address copilot review comment
Jun 19, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 13 additions & 2 deletions cvs/tests/health/transferbench_cvs.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -162,9 +162,20 @@ def parse_tb_scaling_bw(out_dict, exp_dict):

def parse_tb_schmoo_bw(out_dict, exp_dict):
for node in out_dict.keys():
# Line-anchored schmoo data row: leading indent + optional |/│, then whole 32 (not 132/320).
# Use (?<![0-9])32(?![0-9]) instead of \b32\b so indented lines (spaces are non-word) still match.
match = re.search(
'\s+32\s+([0-9\.]+)\s+([0-9\.]+)\s+([0-9\.]+)\s+([0-9\.]+)\s+([0-9\.]+)\s+([0-9\.]+)', out_dict[node]
r'(?m)^[ \t]*(?:[|│][ \t]*)*(?<![0-9])32(?![0-9])(?:[ \t]*[|│])?[ \t]+'
r'([0-9\.]+)[ \t]+([0-9\.]+)[ \t]+([0-9\.]+)[ \t]+([0-9\.]+)[ \t]+([0-9\.]+)[ \t]+([0-9\.]+)',
out_dict[node],
)
Comment on lines 167 to 171
if not match:
out = out_dict[node]
tail = 6000
if len(out) > tail:
out = f"...[truncated {len(out_dict[node]) - tail} chars; schmoo table is usually near the end]...\n{out_dict[node][-tail:]}"
fail_test(f"TransferBench schmoo row for 32 CU not found on node {node}. Output: {out}")
continue
local_read = float(match.group(1))
local_write = float(match.group(2))
local_copy = float(match.group(3))
Expand Down Expand Up @@ -419,7 +430,7 @@ def test_transfer_bench_schmoo(
path = config_dict['path']
rocm_path = detect_rocm_path(orch, config_dict.get('rocm_path', ''))
out_dict = orch.exec(
f"sudo bash -c 'export LD_LIBRARY_PATH={rocm_path}/lib:$LD_LIBRARY_PATH && echo \"LD_LIBRARY_PATH: $LD_LIBRARY_PATH\" && {path}/TransferBench schmoo'",
f"sudo bash -c 'export LD_LIBRARY_PATH={rocm_path}/lib:$LD_LIBRARY_PATH && echo \"LD_LIBRARY_PATH: $LD_LIBRARY_PATH\" && GFX_UNROLL=32 SWEEP_MIN=32 {path}/TransferBench schmoo'",
timeout=(60 * 5),
)
print_test_output(log, out_dict)
Expand Down
Loading