Skip to content
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

Merged
merged 14 commits into from
May 30, 2024
Merged

Conversation

oyilmaz-nvidia
Copy link
Collaborator

What does this PR do ?

Cleaning all of the unused functions, classes, etc, after the TRT-LLM API integration.

@oyilmaz-nvidia oyilmaz-nvidia marked this pull request as ready for review May 22, 2024 04:52
janekl
janekl previously approved these changes May 22, 2024
@github-actions github-actions bot removed the NLP label May 22, 2024
@jiemingz
Copy link
Collaborator

jiemingz commented May 22, 2024

get_tensor_parallel_group is not used anymore.

I think this one too:

@jiemingz
Copy link
Collaborator

jiemingz commented May 22, 2024

I believe cpu_map_location
is duplicated with this

Also it seems to me cpu_map_location and gpu_map_location should be in trt_llm/nemo/nemo.py since thats where they're only used

@jiemingz
Copy link
Collaborator

This function is only used by build and refit so it can be removed, eventually though we'll have to add it back as well when build and refit are moved onto the new API

Copy link
Collaborator

@jiemingz jiemingz left a 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!

@oyilmaz-nvidia
Copy link
Collaborator Author

Thanks @jiemingz for your review. Updating the code per your comments.

@github-actions github-actions bot removed the NLP label May 24, 2024
@oyilmaz-nvidia oyilmaz-nvidia requested review from janekl and removed request for meatybobby and janekl May 30, 2024 18:54
@oyilmaz-nvidia oyilmaz-nvidia merged commit bf53aaa into NVIDIA:main May 30, 2024
130 checks passed
BoxiangW pushed a commit to BoxiangW/NeMo that referenced this pull request Jun 5, 2024
* 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]>
janekl pushed a commit that referenced this pull request Jun 12, 2024
* 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]>
rohitrango pushed a commit to rohitrango/NeMo that referenced this pull request Jun 25, 2024
* 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]>
@ko3n1g ko3n1g mentioned this pull request Jul 18, 2024
2 tasks
XuesongYang pushed a commit to paarthneekhara/NeMo that referenced this pull request Jan 18, 2025
* 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]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants