-
-
Notifications
You must be signed in to change notification settings - Fork 10.6k
[Misc] Move find_loaded_library to platform_aware_utils.py #12231
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
Conversation
👋 Hi! Thank you for contributing to the vLLM project. Once the PR is approved and ready to go, your PR reviewer(s) can run CI to test the changes comprehensively before merging. To run CI, PR reviewers can do one of these:
🚀 |
Signed-off-by: Lu Fang <[email protected]>
515f144
to
f0af1d9
Compare
The change of separating the function makes sense to me. in vllm, can you provide more intuitive names to them? maybe you can provide your implementation of |
How about deployment_aware_utils.py instead? Our internal changes can be found in https://github.com/vllm-project/vllm/pull/12129/files. So far two things to change: 1) libcudart path, 2) PYTHONPATH due to PAR packing method. |
In this case, I think we can have Line 191 in 9691255
|
I see, I can have similar way to handle VLLM_CUDART_SO_PATH. How about the solution for subprocess.run? I was about to have a wrapper of subprocess_run to pass in the PYTHONPATH internally. Any suggestion on that? |
commented at #12129 (comment) |
Hi @houseroad @youkaichao, checking in on this PR, anything blocking landing it? |
Thanks for checking. @youkaichao feels a bit hacky of using file replacement, but it's quite common on the PyTorch internal usage. I am adding another hook in the _version.py internally. Will create some PR to see if it's enough. Let's keep this PR open until the internal hook needs meet our requirement. :-) |
Close this one, since the new proposal got landed. |
find_loaded_library is a platform specific implementation, so create "platform_aware_utils.py". For different platform (such as using PAR to pack the binary), we can create our own version of "platform_aware_utils.py" to override such platform aware utility functions.