Skip to content

Commit

Permalink
Minor fix
Browse files Browse the repository at this point in the history
  • Loading branch information
nv-hwoo committed Oct 26, 2023
1 parent 6d8763d commit f695177
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions src/c++/perf_analyzer/docs/examples/profile.py
Original file line number Diff line number Diff line change
Expand Up @@ -269,9 +269,7 @@ def generate_prompts(args, input_data):
prompt = input_data["data"][0]["PROMPT"][0]

if not prompt: # Generate dummy prompt
assert (
args.prompt_size_range
), "Must specify --prompt-size-range if prompt is not provided."
assert args.prompt_size_range, "Must specify --prompt-size-range."
start, end, step = args.prompt_size_range
return [" ".join(["hi"] * size) for size in range(start, end + 1, step)]
return [prompt]
Expand Down

0 comments on commit f695177

Please sign in to comment.