Enhance runtime memory check in CI (#15192)

test/srt/test_radix_attention.py

@@ -84,4 +84,5 @@ class TestRadixCacheNonOverlapLPM(TestRadixCacheFCFS):
 
 if __name__ == "__main__":
     envs.SGLANG_TEST_RETRACT.set(True)
+    envs.SGLANG_ENABLE_STRICT_MEM_CHECK_DURING_BUSY.set(1)
     unittest.main()

test/srt/test_retract_decode.py

@@ -27,7 +27,9 @@ class TestRetractDecode(CustomTestCase):
         cls.model = DEFAULT_MODEL_NAME_FOR_TEST
         cls.base_url = DEFAULT_URL_FOR_TEST
         launch_args = ["--chunked-prefill-size", "128"] + cls.other_args
-        with envs.SGLANG_TEST_RETRACT.override(True):
+        with envs.SGLANG_TEST_RETRACT.override(
+            True
+        ), envs.SGLANG_ENABLE_STRICT_MEM_CHECK_DURING_BUSY.override(1):
             cls.process = popen_launch_server(
                 cls.model,
                 cls.base_url,