Add e2e test for multi instance multi stage memory release/resume occupuation (#7208)

Signed-off-by: Ata Fatahi <immrata@gmail.com>
2025-06-26 17:40:38 -07:00
parent 3d7cdb2ebd
commit 031f64aa1b
2 changed files with 251 additions and 0 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -160,6 +160,7 @@ suites = {
    "per-commit-4-gpu": [
        TestFile("test_local_attn.py", 250),
        TestFile("test_pp_single_node.py", 150),
+        TestFile("test_multi_instance_release_memory_occupation.py", 64),
    ],
    "per-commit-4-gpu-amd": [
        TestFile("test_pp_single_node.py", 150),