-
Notifications
You must be signed in to change notification settings - Fork 2.4k
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
[AUTO] Implement auto-plugin limited devices feature #5545
[AUTO] Implement auto-plugin limited devices feature #5545
Conversation
9599c0a
to
fb70e8c
Compare
I closed #5549 , let's focus on this PR. Sorry to bring confuse. |
804e29e
to
4e1bed6
Compare
Hi @myshevts @ilya-lavrenov , thanks for your review, I just rebased to the latest master branch.
|
97684e5
to
c59e0b0
Compare
Signed-off-by: Zhengtian Xie <[email protected]>
Signed-off-by: Shoujiang Ma <[email protected]>
Signed-off-by: Shoujiang Ma <[email protected]>
Signed-off-by: Shoujiang Ma <[email protected]>
Signed-off-by: Shoujiang Ma <[email protected]>
Signed-off-by: Shoujiang Ma <[email protected]>
Signed-off-by: Shoujiang Ma <[email protected]>
Signed-off-by: Zhengtian Xie <[email protected]>
Signed-off-by: Shoujiang Ma <[email protected]>
Signed-off-by: Shoujiang Ma <[email protected]>
Signed-off-by: Zhengtian Xie <[email protected]>
Signed-off-by: Shoujiang Ma <[email protected]>
Signed-off-by: Shoujiang Ma <[email protected]>
Signed-off-by: Shoujiang Ma <[email protected]>
Signed-off-by: Shoujiang Ma <[email protected]>
Signed-off-by: Shoujiang Ma <[email protected]>
a4d42bb
to
bce7227
Compare
// and AUTO select CPU but not GPU (GPU has this capability). | ||
if (cap == METRIC_VALUE(BATCHED_BLOB)) { | ||
continue; | ||
} |
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.
looks like we need to respect KEY_AUTO_DEVICE_LIST
here as well. E.i. if only CPU is passed, only CPU optimization capabilities should be taken into account
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.
let's fix in future PR
…#5545) * Update SelectDevice policy in auto plugin Signed-off-by: Zhengtian Xie <[email protected]> * Implement limit device list for AUTO plugin Signed-off-by: Shoujiang Ma <[email protected]> * Add tests for AUTO limit device feature Signed-off-by: Shoujiang Ma <[email protected]> * Add gpu tests for auto-plugin Signed-off-by: Shoujiang Ma <[email protected]> * Fix CI cpuFuncTests issue due to BATCHED_BLOB Signed-off-by: Shoujiang Ma <[email protected]> * Override LoadNetwork(modelPath, config) in AUTO plugin Signed-off-by: Shoujiang Ma <[email protected]> * Update SelectDevice() logic for LoadNetwork(model, config) Signed-off-by: Shoujiang Ma <[email protected]> * Update GetNetworkPrecision logic for auto-plugin Signed-off-by: Zhengtian Xie <[email protected]> * Address reviewers' comments Signed-off-by: Shoujiang Ma <[email protected]> * Add tests for AUTO:GPU,CPU case Signed-off-by: Shoujiang Ma <[email protected]> * Update logic in GetNetworkPrecision for auto-plugin Signed-off-by: Zhengtian Xie <[email protected]> * Address reviewer's comment: clean and simplify code Signed-off-by: Shoujiang Ma <[email protected]> * Fix wrong usage of convolution weight index Signed-off-by: Shoujiang Ma <[email protected]> * Address reviewer comment: fix get network precision logic Signed-off-by: Shoujiang Ma <[email protected]> * Fix rebase issue Signed-off-by: Shoujiang Ma <[email protected]> * Fix ie_core.cpp header change Signed-off-by: Shoujiang Ma <[email protected]> Co-authored-by: zhengtian.xie <[email protected]>
…#5545) * Update SelectDevice policy in auto plugin Signed-off-by: Zhengtian Xie <[email protected]> * Implement limit device list for AUTO plugin Signed-off-by: Shoujiang Ma <[email protected]> * Add tests for AUTO limit device feature Signed-off-by: Shoujiang Ma <[email protected]> * Add gpu tests for auto-plugin Signed-off-by: Shoujiang Ma <[email protected]> * Fix CI cpuFuncTests issue due to BATCHED_BLOB Signed-off-by: Shoujiang Ma <[email protected]> * Override LoadNetwork(modelPath, config) in AUTO plugin Signed-off-by: Shoujiang Ma <[email protected]> * Update SelectDevice() logic for LoadNetwork(model, config) Signed-off-by: Shoujiang Ma <[email protected]> * Update GetNetworkPrecision logic for auto-plugin Signed-off-by: Zhengtian Xie <[email protected]> * Address reviewers' comments Signed-off-by: Shoujiang Ma <[email protected]> * Add tests for AUTO:GPU,CPU case Signed-off-by: Shoujiang Ma <[email protected]> * Update logic in GetNetworkPrecision for auto-plugin Signed-off-by: Zhengtian Xie <[email protected]> * Address reviewer's comment: clean and simplify code Signed-off-by: Shoujiang Ma <[email protected]> * Fix wrong usage of convolution weight index Signed-off-by: Shoujiang Ma <[email protected]> * Address reviewer comment: fix get network precision logic Signed-off-by: Shoujiang Ma <[email protected]> * Fix rebase issue Signed-off-by: Shoujiang Ma <[email protected]> * Fix ie_core.cpp header change Signed-off-by: Shoujiang Ma <[email protected]> Co-authored-by: zhengtian.xie <[email protected]>
Details:
Tickets: