Skip to content

Commit 1e39ef4

Browse files
committed
bring back all tests
Signed-off-by: Balaram Buddharaju <[email protected]>
1 parent a3a1149 commit 1e39ef4

File tree

1 file changed

+6
-18
lines changed

1 file changed

+6
-18
lines changed

cpp/tests/unit_tests/multi_gpu/cacheTransceiverTest.cpp

Lines changed: 6 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1268,7 +1268,7 @@ TEST_P(AsymmetricalCacheTest, TestCase)
12681268
{
12691269
GTEST_SKIP() << "Temporarily skipping cache transceiver tests with NIXL backend for CP.";
12701270
}
1271-
std::vector<int> lenList = {8};
1271+
std::vector<int> lenList = {30, 10, 60, 80};
12721272
if (genCp > 1)
12731273
{
12741274
std::vector<int> updatedLenList;
@@ -1296,7 +1296,7 @@ TEST_P(AsymmetricalCacheTest, TestCase)
12961296
std::vector<std::shared_ptr<WrappedLlmRequest>> requests;
12971297

12981298
// the second loop is for cache reuse
1299-
for (int i = 0; i < 1; i++)
1299+
for (int i = 0; i < 2; i++)
13001300
{
13011301
for (auto len : lenList)
13021302
{
@@ -1473,22 +1473,10 @@ TEST_P(AsymmetricalCacheTestWithDP, TestCase)
14731473
}
14741474

14751475
INSTANTIATE_TEST_CASE_P(AsymmetricCaseTest0, AsymmetricalCacheTest,
1476-
testing::Combine(testing::Values(1),
1477-
testing::Values(1),
1478-
testing::Values(1),
1479-
testing::Values(1),
1480-
testing::Values(1),
1481-
testing::Values(1),
1482-
testing::Values(2),
1483-
testing::Values(1),
1484-
testing::Values(4),
1485-
testing::Values(8),
1486-
testing::Values(nvinfer1::DataType::kINT8),
1487-
testing::Values(1),
1488-
testing::Values(false),
1489-
testing::Values(false),
1490-
testing::Values(false),
1491-
testing::Values(true)));
1476+
testing::Combine(testing::Values(1, 2), testing::Values(1, 2), testing::Values(1), testing::Values(1, 2),
1477+
testing::Values(1, 2), testing::Values(1), testing::Values(4), testing::Values(4), testing::Values(4),
1478+
testing::Values(16), testing::Values(nvinfer1::DataType::kFLOAT, nvinfer1::DataType::kINT8), testing::Values(2),
1479+
testing::Values(false), testing::Values(false), testing::Values(false), testing::Values(true, false)));
14921480

14931481
INSTANTIATE_TEST_CASE_P(AsymmetricCaseTestWithWindow, AsymmetricalCacheTest,
14941482
testing::Combine(testing::Values(1), testing::Values(1), testing::Values(1), testing::Values(1), testing::Values(1),

0 commit comments

Comments
 (0)