Multi Model Server Benchmarking
The benchmarks measure the performance of MMS on various models and benchmarks. It supports either a number of built-in models or a custom model passed in as a path or URL to the .model file. It also runs various benchmarks using these models (see benchmarks section below). The benchmarks are run through a python3 script on the user machine through jmeter. MMS is run on the same machine in a docker instance to avoid network latencies. The benchmark must be run from within the context of the full MMS repo because it executes the local code as the version of MMS (and it is recompiled between runs) for ease of development.
Installation
Ubuntu
The script is mainly intended to run on a Ubuntu EC2 instance. For this reason, we have provided an install_dependencies.sh
script to install everything needed to execute the benchmark on this environment. All you need to do is run this file and clone the MMS repo.
MacOS
For mac, you should have python3 and java installed. If you wish to run the default benchmarks featuring a docker-based instance of MMS, you will need to install docker as well. Finally, you will need to install jmeter with plugins which can be accomplished by running mac_install_dependencies.sh
.
Other
For other environments, manual installation is necessary. The list of dependencies to be installed can be found below or by reading the ubuntu installation script.
The benchmarking script requires the following to run:
- python3
- A JDK and JRE
- jmeter installed through homebrew or linuxbrew with the plugin manager and the following plugins: jpgc-synthesis=2.1,jpgc-filterresults=2.1,jpgc-mergeresults=2.1,jpgc-cmd=2.1,jpgc-perfmon=2.1
- Docker-ce with the current user added to the docker group
- Nvidia-docker (for GPU)
Models
The pre-loaded models for the benchmark can be mostly found in the MMS model zoo. We currently support the following:
- resnet: ResNet-18 (Default)
- squeezenet: SqueezeNet V1.1
- lstm: lstm-ptb
- noop: noop-v1.0 Simple Noop model which returns “Hello world” to any input specified.
- noop_echo: noop_echo-v1.0 Simple Noop model which returns whatever input is given to it.
Benchmarks
We support several basic benchmarks:
- throughput: Run inference with enough threads to occupy all workers and ensure full saturation of resources to find the throughput. The number of threads defaults to 100.
- latency: Run inference with a single thread to determine the latency
- ping: Test the throughput of pinging against the frontend
- load: Loads the same model many times in parallel. The number of loads is given by the “count” option and defaults to 16.
- repeated_scale_calls: Will scale the model up to “scale_up_workers”=16 then down to “scale_down_workers”=1 then up and down repeatedly.
- multiple_models: Loads and scales up three models (1. noop, 2. lstm, and 3. resnet), at the same time, runs inferences on them, and then scales them down. Use the options “urlN”, “modelN_name”, “dataN” to specify the model url, model name, and the data to pass to the model respectively. data1 and data2 are of the format “'Some garbage data being passed here'” and data3 is the filesystem path to a file to upload.
We also support compound benchmarks:
- concurrent_inference: Runs the basic benchmark with different numbers of threads
Examples
Run basic latency test on default resnet-18 model
./benchmark.py latency
Run basic throughput test on default resnet-18 model.
./benchmark.py throughput
Run all benchmarks
./benchmark.py --all
Run using the noop-v1.0 model
./benchmark.py latency -m noop_v1.0
Run on GPU (4 gpus)
./benchmark.py latency -g 4
Run with a custom image
./benchmark.py latency -i {imageFilePath}
Run with a custom model (works only for CNN based models, which accept image as an input for now. We will add support for more input types in future to this command. )
./benchmark.py latency -c {modelUrl} -i {imageFilePath}
Run with custom options
./benchmark.py repeated_scale_calls --options scale_up_workers 100 scale_down_workers 10
Run against an already running instance of MMS
./benchmark.py latency --mms 127.0.0.1
(defaults to http, port 80, management port = port + 1)
./benchmark.py latency --mms 127.0.0.1:8080 --management-port 8081
./benchmark.py latency --mms https://127.0.0.1:8443
Run verbose with only a single loop
./benchmark.py latency -v -l 1
Benchmark options
The full list of options can be found by running with the -h or –help flags.
Profiling
Frontend
The benchmarks can be used in conjunction with standard profiling tools such as JProfiler to analyze the system performance. JProfiler can be downloaded from their website. Once downloaded, open up JProfiler and follow these steps:
- Run MMS directly through gradle (do not use docker). This can be done either on your machine or on a remote machine accessible through SSH.
- In JProfiler, select “Attach” from the ribbon and attach to the ModelServer. The process name in the attach window should be “com.amazonaws.ml.mms.ModelServer”. If it is on a remote machine, select “On another computer” in the attach window and enter the SSH details. For the session startup settings, you can leave it with the defaults. At this point, you should see live CPU and Memory Usage data on JProfiler’s Telemetries section.
- Select Start Recordings in JProfiler’s ribbon
- Run the Benchmark script targeting your running MMS instance. It might run something like
./benchmark.py throughput --mms https://127.0.0.1:8443
. It can be run on either your local machine or a remote machine (if you are running remote), but we recommend running the benchmark on the same machine as the model server to avoid confounding network latencies. - Once the benchmark script has finished running, select Stop Recordings in JProfiler’s ribbon
Once you have stopped recording, you should be able to analyze the data. One useful section to examine is CPU views > Call Tree and CPU views > Hot Spots to see where the processor time is going.
Backend
The benchmarks can also be used to analyze the backend performance using cProfile. It does not require any additional packages to run the benchmark, but viewing the logs does require an additional package. Run pip install snakeviz
to install this. To run the python profiling, follow these steps:
- In the file
mms/model_service_worker.py
, set the constant BENCHMARK to true at the top to enable benchmarking. - Run the benchmark and MMS. They can either be done automatically inside the docker container or separately with the “–mms” flag.
- Run MMS directly through gradle (do not use docker). This can be done either on your machine or on a remote machine accessible through SSH.
- Run the Benchmark script targeting your running MMS instance. It might run something like
./benchmark.py throughput --mms https://127.0.0.1:8443
. It can be run on either your local machine or a remote machine (if you are running remote), but we recommend running the benchmark on the same machine as the model server to avoid confounding network latencies. - Run
snakeviz /tmp/mmsPythonProfile.prof
to view the profiling data. It should start up a web server on your machine and automatically open the page. - Don’t forget to set BENCHMARK = False in the model_service_worker.py file after you are finished.