20 lines
602 B
Python
20 lines
602 B
Python
import os
|
|
|
|
from vllm.compilation.levels import CompilationLevel
|
|
|
|
from ..utils import compare_two_settings
|
|
|
|
# --enforce-eager on TPU causes graph compilation
|
|
# this times out default Health Check in the MQLLMEngine,
|
|
# so we set the timeout here to 30s
|
|
os.environ["VLLM_RPC_TIMEOUT"] = "30000"
|
|
|
|
|
|
def test_custom_dispatcher():
|
|
compare_two_settings(
|
|
"google/gemma-2b",
|
|
arg1=["--enforce-eager"],
|
|
arg2=["--enforce-eager"],
|
|
env1={"VLLM_TORCH_COMPILE_LEVEL": str(CompilationLevel.DYNAMO_ONCE)},
|
|
env2={"VLLM_TORCH_COMPILE_LEVEL": str(CompilationLevel.DYNAMO_AS_IS)})
|