-
Notifications
You must be signed in to change notification settings - Fork 2.9k
Adding more export tests #13410
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Adding more export tests #13410
Conversation
Signed-off-by: Onur Yilmaz <[email protected]>
Signed-off-by: Onur Yilmaz <[email protected]>
Signed-off-by: Onur Yilmaz <[email protected]>
Signed-off-by: Onur Yilmaz <[email protected]>
Signed-off-by: Onur Yilmaz <[email protected]>
Signed-off-by: Onur Yilmaz <[email protected]>
Signed-off-by: oyilmaz-nvidia <[email protected]>
[🤖]: Hi @oyilmaz-nvidia 👋, We wanted to let you know that a CICD pipeline for this PR just finished successfully. So it might be time to merge this PR or get some approvals. Due to a major CI change, merges are currently handled by the automation team. |
Signed-off-by: Onur Yilmaz <[email protected]>
…IA/NeMo into onur/more-more-export-tests
Signed-off-by: oyilmaz-nvidia <[email protected]>
Signed-off-by: Onur Yilmaz <[email protected]>
…IA/NeMo into onur/more-more-export-tests
Signed-off-by: Onur Yilmaz <[email protected]>
Signed-off-by: Onur Yilmaz <[email protected]>
…IA/NeMo into onur/more-more-export-tests
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Last commit only moves those tests to GPU and removes critical import. Should be fine to fast-merge
* Tests for vLLMHFExporter and TensorRT lazy compiler Signed-off-by: Onur Yilmaz <[email protected]> * Adding sentencepiece tokenizer test for export Signed-off-by: Onur Yilmaz <[email protected]> * More unit tests for tensorrt_llm.py Signed-off-by: Onur Yilmaz <[email protected]> * more tensorrt_mm_exporter tests Signed-off-by: Onur Yilmaz <[email protected]> * Fix style Signed-off-by: Onur Yilmaz <[email protected]> * Add header Signed-off-by: Onur Yilmaz <[email protected]> * Apply isort and black reformatting Signed-off-by: oyilmaz-nvidia <[email protected]> * new tests for lora and export utils Signed-off-by: Onur Yilmaz <[email protected]> * Apply isort and black reformatting Signed-off-by: oyilmaz-nvidia <[email protected]> * Fix some of the codeql issues Signed-off-by: Onur Yilmaz <[email protected]> * Add header Signed-off-by: Onur Yilmaz <[email protected]> * Marking tests as GPU only Signed-off-by: Onur Yilmaz <[email protected]> --------- Signed-off-by: Onur Yilmaz <[email protected]> Signed-off-by: oyilmaz-nvidia <[email protected]> Co-authored-by: oyilmaz-nvidia <[email protected]> Co-authored-by: oliver könig <[email protected]>
* Tests for vLLMHFExporter and TensorRT lazy compiler Signed-off-by: Onur Yilmaz <[email protected]> * Adding sentencepiece tokenizer test for export Signed-off-by: Onur Yilmaz <[email protected]> * More unit tests for tensorrt_llm.py Signed-off-by: Onur Yilmaz <[email protected]> * more tensorrt_mm_exporter tests Signed-off-by: Onur Yilmaz <[email protected]> * Fix style Signed-off-by: Onur Yilmaz <[email protected]> * Add header Signed-off-by: Onur Yilmaz <[email protected]> * Apply isort and black reformatting Signed-off-by: oyilmaz-nvidia <[email protected]> * new tests for lora and export utils Signed-off-by: Onur Yilmaz <[email protected]> * Apply isort and black reformatting Signed-off-by: oyilmaz-nvidia <[email protected]> * Fix some of the codeql issues Signed-off-by: Onur Yilmaz <[email protected]> * Add header Signed-off-by: Onur Yilmaz <[email protected]> * Marking tests as GPU only Signed-off-by: Onur Yilmaz <[email protected]> --------- Signed-off-by: Onur Yilmaz <[email protected]> Signed-off-by: oyilmaz-nvidia <[email protected]> Co-authored-by: oyilmaz-nvidia <[email protected]> Co-authored-by: oliver könig <[email protected]> Signed-off-by: Charlie Truong <[email protected]>
* Tests for vLLMHFExporter and TensorRT lazy compiler * Adding sentencepiece tokenizer test for export * More unit tests for tensorrt_llm.py * more tensorrt_mm_exporter tests * Fix style * Add header * Apply isort and black reformatting * new tests for lora and export utils * Apply isort and black reformatting * Fix some of the codeql issues * Add header * Marking tests as GPU only --------- Signed-off-by: Onur Yilmaz <[email protected]> Signed-off-by: oyilmaz-nvidia <[email protected]> Co-authored-by: Onur Yilmaz <[email protected]> Co-authored-by: oyilmaz-nvidia <[email protected]>
Important
The
Update branch
button must only be pressed in very rare occassions.An outdated branch is never blocking the merge of a PR.
Please reach out to the automation team before pressing that button.
What does this PR do ?
Adds more export tests to get more coverage.