Unit test for Hierarchical Caching (#4486)

2025-03-17 17:45:00 -07:00
parent 9b81f9bd34
commit a98290aea3
7 changed files with 65 additions and 5 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -74,6 +74,8 @@ suites = {
        TestFile("test_w8a8_quantization.py", 46),
        TestFile("test_eval_fp8_accuracy.py", 172),
        TestFile("test_create_kvindices.py", 2),
+        TestFile("test_hicache.py", 60),
+        TestFile("test_hicache_mla.py", 90),
    ],
    "nightly": [
        TestFile("test_nightly_gsm8k_eval.py"),