Skip to content

Commit bcfd4c4

Browse files
jpascucci-nvjpascucci-nv
committed
fix(cuda.core): serialize memory-pool watermark reset tests
Mark test_mempool_watermark_reset thread_unsafe, matching test_mempool_trim. Both exercise the pool teardown and synchronization sequence associated with the suspected driver race in #3350.
1 parent 587c111 commit bcfd4c4

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

‎cuda_core/tests/test_memory.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2973,6 +2973,7 @@ def test_mempool_release_threshold_accepts_index_types(memory_resource_factory,
29732973
mr.close()
29742974

29752975

2976+
@pytest.mark.thread_unsafe(reason="CUDA legacy-default-stream mempool teardown/sync race (gh-3350)")
29762977
@pytest.mark.agent_authored(model="claude-opus-5-5")
29772978
@pytest.mark.parametrize("kind", ["reserved", "used"])
29782979
def test_mempool_watermark_reset(memory_resource_factory, kind):

0 commit comments

Comments
 (0)