diff --git a/shortfin/tests/apps/llm/components/kvcache/base_attention_cache_test.py b/shortfin/tests/apps/llm/components/kvcache/base_attention_cache_test.py index 113da6912..181cc8d9f 100644 --- a/shortfin/tests/apps/llm/components/kvcache/base_attention_cache_test.py +++ b/shortfin/tests/apps/llm/components/kvcache/base_attention_cache_test.py @@ -32,7 +32,7 @@ def acquire_free_pages(self, count: int) -> List[PageInfo]: except queue.Empty: return None - def release_pages(self, pages): + def free_pages(self, pages): for page in pages: self._queue.put(page)