Skip to content

Commit c5cc688

Browse files
committed
test: Add logging for beam width in executor tests
- Introduced logging to track the current beam width during token comparison tests, enhancing test traceability. Signed-off-by: Robin Kobus <[email protected]>
1 parent 0e2c929 commit c5cc688

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

cpp/tests/e2e_tests/executor/executorTest.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2459,6 +2459,7 @@ void doTokenComparisonChangeBeamWidth(bool enableReuse, SizeType32 maxWaitMs)
24592459

24602460
for (SizeType32 beamWidth : {1, 2})
24612461
{
2462+
TLLM_LOG_INFO("Running beam width: %d", beamWidth);
24622463
BeamResult beamResult{beamWidth};
24632464
auto const resultsPath
24642465
= GPT_DATA_PATH / ((beamWidth == 1) ? "sampling" : "beam_search_" + std::to_string(beamWidth));

0 commit comments

Comments
 (0)