From 62d3e97618fcfc23ae562a3f91d6dfd38351fc03 Mon Sep 17 00:00:00 2001 From: Cedar Date: Mon, 2 Dec 2024 11:29:39 -0800 Subject: [PATCH] fix missing rename --- .../apps/llm/components/kvcache/base_attention_cache_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)