Skip to content

Commit a79f641

Browse files
committed
chore: Remove static batching from ExecutorTest
- Updated `validateContextLogits` and `validateGenerationLogits` functions to remove the `batchingType` parameter. - Adjusted related test functions to reflect the changes in parameter lists. - Cleaned up the instantiation of test cases to eliminate unnecessary batchingType references. Signed-off-by: Robin Kobus <[email protected]>
1 parent 9b6c51d commit a79f641

File tree

4 files changed

+141
-114
lines changed

4 files changed

+141
-114
lines changed

cpp/tests/executor/disaggExecutorTest.cpp

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -251,11 +251,10 @@ void runDisaggTest(tensorrt_llm::testing::disaggexecutor::DisaggExecutorLeader&
251251
flakyTestInfo);
252252

253253
testData.validateContextLogits(outConfig.returnContextLogits, givenInputLengths.at(batchId),
254-
beamWidth, contextLogits, vocabSizePadded, batchId, batchingType);
254+
beamWidth, contextLogits, vocabSizePadded, batchId);
255255
testData.validateGenerationLogits(outConfig.returnGenerationLogits, result.isFinal, streaming,
256256
outConfig.excludeInputFromOutput, givenInputLengths.at(batchId), reqMaxNewTokens.at(batchId),
257-
beamWidth, beamTokens, genLogits, vocabSizePadded, batchId, batchingType,
258-
returnAllGeneratedTokens);
257+
beamWidth, beamTokens, genLogits, vocabSizePadded, batchId, returnAllGeneratedTokens);
259258
}
260259
else
261260
{
@@ -410,11 +409,10 @@ void runDisaggTest(DisaggExecutorOrchestrator& executor, tensorrt_llm::runtime::
410409
flakyTestInfo);
411410

412411
testData.validateContextLogits(outConfig.returnContextLogits, givenInputLengths.at(batchId),
413-
beamWidth, contextLogits, vocabSizePadded, batchId, batchingType);
412+
beamWidth, contextLogits, vocabSizePadded, batchId);
414413
testData.validateGenerationLogits(outConfig.returnGenerationLogits, result.isFinal, streaming,
415414
outConfig.excludeInputFromOutput, givenInputLengths.at(batchId), reqMaxNewTokens.at(batchId),
416-
beamWidth, beamTokens, genLogits, vocabSizePadded, batchId, batchingType,
417-
returnAllGeneratedTokens);
415+
beamWidth, beamTokens, genLogits, vocabSizePadded, batchId, returnAllGeneratedTokens);
418416
}
419417
else
420418
{

0 commit comments

Comments
 (0)