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 and others added 5 commits May 21, 2024 17:02
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>
@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
Contributor

jiemingz commented May 22, 2024

get_tensor_parallel_group is not used anymore.

I think this one too:

@jiemingz
Copy link
Contributor

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
Contributor

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
Contributor

@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!

Signed-off-by: oyilmaz-nvidia <oyilmaz-nvidia@users.noreply.github.com>
@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
Signed-off-by: Onur Yilmaz <oyilmaz@nvidia.com>
@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 <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>
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 <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>
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 <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>
@ko3n1g ko3n1g mentioned this pull request Jul 18, 2024
2 tasks
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