-
Notifications
You must be signed in to change notification settings - Fork 2.5k
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
TRT-LLM Export Code Cleanup #9270
Conversation
Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com>
Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com>
Signed-off-by: oyilmaz-nvidia <oyilmaz-nvidia@users.noreply.github.com>
Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com>
… into export-cleanup
get_tensor_parallel_group is not used anymore. I think this one too: |
I believe cpu_map_location Also it seems to me |
This function is only used by |
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.
Please check my comments above @oyilmaz-nvidia, thanks!
Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com>
… into export-cleanup
Signed-off-by: oyilmaz-nvidia <oyilmaz-nvidia@users.noreply.github.com>
Thanks @jiemingz for your review. Updating the code per your comments. |
Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com>
… into export-cleanup
* Init code cleanup for the trt-llm export Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com> * Removed model config Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com> * Apply isort and black reformatting Signed-off-by: oyilmaz-nvidia <oyilmaz-nvidia@users.noreply.github.com> * clearn futher Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com> * Done more cleaning and addressing the reviews Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com> * Apply isort and black reformatting Signed-off-by: oyilmaz-nvidia <oyilmaz-nvidia@users.noreply.github.com> --------- Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com> Signed-off-by: oyilmaz-nvidia <oyilmaz-nvidia@users.noreply.github.com> Co-authored-by: oyilmaz-nvidia <oyilmaz-nvidia@users.noreply.github.com> Signed-off-by: Boxiang Wang <boxiangw@nvidia.com>
* Init code cleanup for the trt-llm export Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com> * Removed model config Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com> * Apply isort and black reformatting Signed-off-by: oyilmaz-nvidia <oyilmaz-nvidia@users.noreply.github.com> * clearn futher Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com> * Done more cleaning and addressing the reviews Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com> * Apply isort and black reformatting Signed-off-by: oyilmaz-nvidia <oyilmaz-nvidia@users.noreply.github.com> --------- Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com> Signed-off-by: oyilmaz-nvidia <oyilmaz-nvidia@users.noreply.github.com> Co-authored-by: oyilmaz-nvidia <oyilmaz-nvidia@users.noreply.github.com> Signed-off-by: Jan Lasek <janek.lasek@gmail.com>
* Init code cleanup for the trt-llm export Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com> * Removed model config Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com> * Apply isort and black reformatting Signed-off-by: oyilmaz-nvidia <oyilmaz-nvidia@users.noreply.github.com> * clearn futher Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com> * Done more cleaning and addressing the reviews Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com> * Apply isort and black reformatting Signed-off-by: oyilmaz-nvidia <oyilmaz-nvidia@users.noreply.github.com> --------- Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com> Signed-off-by: oyilmaz-nvidia <oyilmaz-nvidia@users.noreply.github.com> Co-authored-by: oyilmaz-nvidia <oyilmaz-nvidia@users.noreply.github.com>
What does this PR do ?
Cleaning all of the unused functions, classes, etc, after the TRT-LLM API integration.