@@ -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
14751475INSTANTIATE_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
14931481INSTANTIATE_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