Skip to content

Refactored LLM benchmark code#899

Merged
svaruag merged 19 commits into
masterfrom
mrwyattii/refactor-inf-bench-classes
Jun 26, 2024
Merged

Refactored LLM benchmark code#899
svaruag merged 19 commits into
masterfrom
mrwyattii/refactor-inf-bench-classes

Conversation

@mrwyattii
Copy link
Copy Markdown
Contributor

No description provided.

Comment thread benchmarks/inference/llm-bench/src/llm_bench/prompt.py Outdated
Comment thread benchmarks/inference/llm-bench/src/llm_bench/prompt.py Outdated
Comment thread benchmarks/inference/llm-bench/src/llm_bench/arg_parsing.py Outdated
Comment thread benchmarks/inference/llm-bench/src/llm_bench/benchmark_runner.py
Comment thread benchmarks/inference/llm-bench/src/llm_bench/benchmark_runner.py Outdated
Comment thread benchmarks/inference/llm-bench/src/llm_bench/benchmark_runner.py
Comment thread benchmarks/inference/llm-bench/configs/1300-120.yaml
Copy link
Copy Markdown
Contributor

@lekurile lekurile left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good to me! Left a few comments.

Comment thread benchmarks/inference/llm-bench/configs/128k-120.yaml Outdated
@mrwyattii mrwyattii marked this pull request as ready for review June 21, 2024 17:55
@svaruag svaruag merged commit bbab278 into master Jun 26, 2024
hwchen2017 pushed a commit that referenced this pull request Jun 8, 2025
* add refactored LLM benchmark code, initial commit

* move prompt processing outside benchmark loop

* formatting and improvements

* slight refactor of benchmark runner, cleaning up, adding type hints

* add tests, small refactors to improve code, make installable package

* clean up code, add TODO notes for intended changes

* update author list

* add early stopping of benchmarks

* add support for longer prompts and cleanup

* address PR feedback, fix bugs, small updates

* small change

* fix small bugs around prompt length and max token size

* remove debug prints

* Update 128k-120.yaml

* add min_requests override and print out for result summary

* add README, rename benchmark

* update unit tests
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants