diff --git a/test/registered/hicache/test_hicache_variants.py b/test/registered/hicache/test_hicache_variants.py index 419ef37f0..1f130dd2d 100644 --- a/test/registered/hicache/test_hicache_variants.py +++ b/test/registered/hicache/test_hicache_variants.py @@ -81,6 +81,8 @@ class TestHiCacheMLA(HiCacheBaseServer, MMLUMixin, MGSMEnMixin): mmlu_num_examples = 256 mmlu_num_threads = 32 mgsm_en_score_threshold = 0.8 + if _is_hip: + mgsm_en_num_threads = 32 @unittest.skipIf(is_hip(), "Disabled for AMD-aiter")