From 52f00eb828ffcffd73055a934156f601d98153e3 Mon Sep 17 00:00:00 2001 From: "liyangcheng.lyc" Date: Fri, 24 Jan 2025 16:52:58 +0800 Subject: [PATCH 1/2] [Flashinfer][Fix] fix missing args in flashinfer test --- .../test_runtime_builtin_paged_attention_kv_cache_flashinfer.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/python/relax/test_runtime_builtin_paged_attention_kv_cache_flashinfer.py b/tests/python/relax/test_runtime_builtin_paged_attention_kv_cache_flashinfer.py index 4c25383178ac..0680be900e65 100644 --- a/tests/python/relax/test_runtime_builtin_paged_attention_kv_cache_flashinfer.py +++ b/tests/python/relax/test_runtime_builtin_paged_attention_kv_cache_flashinfer.py @@ -382,6 +382,8 @@ def create_kv_cache(rope_mode): None, None, None, + None, + False ) return cache From 9fee72e3ee3b25c1b3c840c0338d0cd29d1aad3c Mon Sep 17 00:00:00 2001 From: Ruihang Lai Date: Sat, 25 Jan 2025 18:10:32 -0500 Subject: [PATCH 2/2] Fix lint issue --- .../test_runtime_builtin_paged_attention_kv_cache_flashinfer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/relax/test_runtime_builtin_paged_attention_kv_cache_flashinfer.py b/tests/python/relax/test_runtime_builtin_paged_attention_kv_cache_flashinfer.py index 0680be900e65..2422b16f5a04 100644 --- a/tests/python/relax/test_runtime_builtin_paged_attention_kv_cache_flashinfer.py +++ b/tests/python/relax/test_runtime_builtin_paged_attention_kv_cache_flashinfer.py @@ -383,7 +383,7 @@ def create_kv_cache(rope_mode): None, None, None, - False + False, ) return cache