"...gaoqiong/composable_kernel.git" did not exist on "30e10f8046c8d62110b6032ca479b1407376299f"
Unverified Commit 1e37c103 authored by Keiven C's avatar Keiven C Committed by GitHub
Browse files

fix: add timeout to tests before lora to unblock (#4808)


Signed-off-by: default avatarKeiven Chang <keivenchang@users.noreply.github.com>
Co-authored-by: default avatarKeiven Chang <keivenchang@users.noreply.github.com>
parent 4ca1679c
...@@ -36,8 +36,8 @@ async def distributed_runtime(): ...@@ -36,8 +36,8 @@ async def distributed_runtime():
runtime.shutdown() runtime.shutdown()
@pytest.mark.asyncio @pytest.mark.timeout(5) # Expected: ~1s, timeout set to 5x for safety
async def test_radix_tree_binding(distributed_runtime): def test_radix_tree_binding():
"""Test RadixTree binding directly with store event and find matches""" """Test RadixTree binding directly with store event and find matches"""
import json import json
...@@ -102,13 +102,12 @@ async def test_radix_tree_binding(distributed_runtime): ...@@ -102,13 +102,12 @@ async def test_radix_tree_binding(distributed_runtime):
) )
@pytest.mark.asyncio @pytest.mark.timeout(5) # Expected: ~1s, timeout set to 5x for safety
@pytest.mark.parametrize("num_threads", [2, 3, 5, 128]) @pytest.mark.parametrize("num_threads", [2, 3, 5, 128])
@pytest.mark.parametrize("prepopulate_worker_ids", [True, False]) @pytest.mark.parametrize("prepopulate_worker_ids", [True, False])
@pytest.mark.parametrize("expiration_duration_secs", [None]) @pytest.mark.parametrize("expiration_duration_secs", [None])
@pytest.mark.parametrize("is_threaded", [True, False]) @pytest.mark.parametrize("is_threaded", [True, False])
async def test_radix_tree_thread_safety( def test_radix_tree_thread_safety(
distributed_runtime,
num_threads, num_threads,
prepopulate_worker_ids, prepopulate_worker_ids,
expiration_duration_secs, expiration_duration_secs,
...@@ -205,6 +204,7 @@ async def test_radix_tree_thread_safety( ...@@ -205,6 +204,7 @@ async def test_radix_tree_thread_safety(
@pytest.mark.asyncio @pytest.mark.asyncio
@pytest.mark.timeout(5) # Expected: ~1s, timeout set to 5x for safety
async def test_event_handler(distributed_runtime): async def test_event_handler(distributed_runtime):
kv_block_size = 32 kv_block_size = 32
namespace = "kv_test" namespace = "kv_test"
...@@ -247,6 +247,7 @@ async def test_event_handler(distributed_runtime): ...@@ -247,6 +247,7 @@ async def test_event_handler(distributed_runtime):
@pytest.mark.asyncio @pytest.mark.asyncio
@pytest.mark.timeout(5) # Expected: ~1s, timeout set to 5x for safety
async def test_approx_kv_indexer(distributed_runtime): async def test_approx_kv_indexer(distributed_runtime):
"""Test ApproxKvIndexer with TTL-based block tracking""" """Test ApproxKvIndexer with TTL-based block tracking"""
kv_block_size = 32 kv_block_size = 32
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment