-
Notifications
You must be signed in to change notification settings - Fork 2.7k
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 <[email protected]>
Signed-off-by: Onur Yilmaz <[email protected]>
Signed-off-by: oyilmaz-nvidia <[email protected]>
Signed-off-by: Onur Yilmaz <[email protected]>
… 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 <[email protected]>
… into export-cleanup
Signed-off-by: oyilmaz-nvidia <[email protected]>
Thanks @jiemingz for your review. Updating the code per your comments. |
Signed-off-by: Onur Yilmaz <[email protected]>
… into export-cleanup
* Init code cleanup for the trt-llm export Signed-off-by: Onur Yilmaz <[email protected]> * Removed model config Signed-off-by: Onur Yilmaz <[email protected]> * Apply isort and black reformatting Signed-off-by: oyilmaz-nvidia <[email protected]> * clearn futher Signed-off-by: Onur Yilmaz <[email protected]> * Done more cleaning and addressing the reviews Signed-off-by: Onur Yilmaz <[email protected]> * Apply isort and black reformatting Signed-off-by: oyilmaz-nvidia <[email protected]> --------- Signed-off-by: Onur Yilmaz <[email protected]> Signed-off-by: oyilmaz-nvidia <[email protected]> Co-authored-by: oyilmaz-nvidia <[email protected]> Signed-off-by: Boxiang Wang <[email protected]>
* Init code cleanup for the trt-llm export Signed-off-by: Onur Yilmaz <[email protected]> * Removed model config Signed-off-by: Onur Yilmaz <[email protected]> * Apply isort and black reformatting Signed-off-by: oyilmaz-nvidia <[email protected]> * clearn futher Signed-off-by: Onur Yilmaz <[email protected]> * Done more cleaning and addressing the reviews Signed-off-by: Onur Yilmaz <[email protected]> * Apply isort and black reformatting Signed-off-by: oyilmaz-nvidia <[email protected]> --------- Signed-off-by: Onur Yilmaz <[email protected]> Signed-off-by: oyilmaz-nvidia <[email protected]> Co-authored-by: oyilmaz-nvidia <[email protected]> Signed-off-by: Jan Lasek <[email protected]>
* Init code cleanup for the trt-llm export Signed-off-by: Onur Yilmaz <[email protected]> * Removed model config Signed-off-by: Onur Yilmaz <[email protected]> * Apply isort and black reformatting Signed-off-by: oyilmaz-nvidia <[email protected]> * clearn futher Signed-off-by: Onur Yilmaz <[email protected]> * Done more cleaning and addressing the reviews Signed-off-by: Onur Yilmaz <[email protected]> * Apply isort and black reformatting Signed-off-by: oyilmaz-nvidia <[email protected]> --------- Signed-off-by: Onur Yilmaz <[email protected]> Signed-off-by: oyilmaz-nvidia <[email protected]> Co-authored-by: oyilmaz-nvidia <[email protected]>
* Init code cleanup for the trt-llm export Signed-off-by: Onur Yilmaz <[email protected]> * Removed model config Signed-off-by: Onur Yilmaz <[email protected]> * Apply isort and black reformatting Signed-off-by: oyilmaz-nvidia <[email protected]> * clearn futher Signed-off-by: Onur Yilmaz <[email protected]> * Done more cleaning and addressing the reviews Signed-off-by: Onur Yilmaz <[email protected]> * Apply isort and black reformatting Signed-off-by: oyilmaz-nvidia <[email protected]> --------- Signed-off-by: Onur Yilmaz <[email protected]> Signed-off-by: oyilmaz-nvidia <[email protected]> Co-authored-by: oyilmaz-nvidia <[email protected]>
What does this PR do ?
Cleaning all of the unused functions, classes, etc, after the TRT-LLM API integration.