-
Notifications
You must be signed in to change notification settings - Fork 205
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
Improve trim tensor implementation #423
Improve trim tensor implementation #423
Conversation
@@ -459,6 +469,7 @@ jobs: | |||
source ./ov/setupvars.sh | |||
python -m pip install --upgrade-strategy eager -r ./text_generation/causal_lm/cpp/requirements.txt | |||
python -m pip install ./thirdparty/openvino_tokenizers/[transformers] | |||
sudo apt-get install libtbb-dev |
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.
maybe we can install TBB only when it's required?
and in cmake we will check:
find_package(TBB COMPONENTS tbb)
if(TBB_FOUND)
target_link_libraries(speculative_decoding_lm PRIVATE TBB::tbb)
endif()
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.
LGTM in general
BTW, it's a temporary change until we merge and release openvinotoolkit/openvino#24359 |
@sammysun0711 , your trim tensor improvement proposal has been merged. Thank you! |
Ticket: 140109