Skip to content

Commit

Permalink
Raise error when unknown backend specified.
Browse files Browse the repository at this point in the history
  • Loading branch information
nv-hwoo committed Dec 1, 2023
1 parent a5b5ae4 commit 7205793
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions src/c++/perf_analyzer/docs/examples/profile.py
Original file line number Diff line number Diff line change
Expand Up @@ -571,6 +571,11 @@ def main(args):
input_data = construct_trtllm_input_data(args)
elif args.backend in "vllm":
input_data = construct_vllm_input_data(args)
else:
raise ValueError(
"Unknown backend specified. Supported backend types are: 'trtllm' "
"and 'vllm'."
)

prompts = generate_prompts(args, input_data)

Expand Down

0 comments on commit 7205793

Please sign in to comment.