Skip to content

vllm.entrypoints.cli.benchmark.base

Classes:

BenchmarkSubcommandBase

Bases: CLISubcommand

The base class of subcommands for vllm bench.

Methods:

  • add_cli_args –

    Add the CLI arguments to the parser.

  • cmd –

    Run the benchmark.

Source code in vllm/entrypoints/cli/benchmark/base.py 
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
class BenchmarkSubcommandBase(CLISubcommand):
    """The base class of subcommands for `vllm bench`."""

    help: str

    @classmethod
    def add_cli_args(cls, parser: argparse.ArgumentParser) -> None:
        """Add the CLI arguments to the parser."""
        raise NotImplementedError

    @staticmethod
    def cmd(args: argparse.Namespace) -> None:
        """Run the benchmark.

        Args:
            args: The arguments to the command.
        """
        raise NotImplementedError

add_cli_args(parser) classmethod

Add the CLI arguments to the parser.

Source code in vllm/entrypoints/cli/benchmark/base.py 
13
14
15
16
@classmethod
def add_cli_args(cls, parser: argparse.ArgumentParser) -> None:
    """Add the CLI arguments to the parser."""
    raise NotImplementedError

cmd(args) staticmethod

Run the benchmark.

Parameters:

  • args

    (Namespace) –

    The arguments to the command.

Source code in vllm/entrypoints/cli/benchmark/base.py 
18
19
20
21
22
23
24
25
@staticmethod
def cmd(args: argparse.Namespace) -> None:
    """Run the benchmark.

    Args:
        args: The arguments to the command.
    """
    raise NotImplementedError