Skip to content

Commit b22488d

Browse files
HJSangclaude
andcommitted
test(gpu): make sleep/wake test memory util env-configurable (GMU)
Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com> Signed-off-by: Hejian Sang <sanghj0923@gmail.com>
1 parent 1918f30 commit b22488d

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

test/runtime/test_sleep_wakeup_gpu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def main() -> None:
4343
engine = Engine(
4444
model=model,
4545
enable_memory_saver=True,
46-
gpu_memory_utilization=0.1,
46+
gpu_memory_utilization=float(os.environ.get("GMU", "0.1")),
4747
max_model_len=2048,
4848
trust_remote_code=True,
4949
log_level="info",

0 commit comments

Comments
 (0)