test_custom_wrapper.py 279 B

1234567891011
  1. from ..utils import compare_two_settings
  2. def test_custom_dispatcher():
  3. compare_two_settings(
  4. "google/gemma-2b",
  5. arg1=["--enforce-eager"],
  6. arg2=["--enforce-eager"],
  7. env1={"APHRODITE_DYNAMO_USE_CUSTOM_DISPATCHER": "0"},
  8. env2={},
  9. )