IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[GPU] updated to add reorder after parameter when it is connected to quantize (#19182)

Browse Source 
* updated to add reorder after parameter when it is connected to quantize

* removed redundant code
This commit is contained in:
Eddy Kim 2023-08-15 15:29:49 +09:00 committed by GitHub
parent 37a6848f26
commit 0194cdce3b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/plugins/intel_gpu/src/plugin/ops/parameter.cpp
View File

@ -162,6 +162,15 @@ static void CreateParameterOp(Program& p, const std::shared_ptr<ngraph::op::v0::
return surface_input_found; return surface_input_found;
}; };
std::function<bool(const std::shared_ptr<ov::Node>&)> connected_to_quantize =
[&](const std::shared_ptr<ov::Node> &node) -> bool {
for (auto& user : node->get_users()) {
if (ngraph::is_type<ngraph::op::v0::FakeQuantize>(user))
return true;
}
return false;
};
size_t search_depth = 3; size_t search_depth = 3;
bool is_convert_color_input = recursive_search_convert_color(op, search_depth); bool is_convert_color_input = recursive_search_convert_color(op, search_depth);
bool is_surface_input = has_surface_input(op); bool is_surface_input = has_surface_input(op);
@ -215,7 +224,7 @@ static void CreateParameterOp(Program& p, const std::shared_ptr<ngraph::op::v0::
switch (preProcess.getMeanVariant()) { switch (preProcess.getMeanVariant()) {
case NONE: { case NONE: {
// If mean value is not specified and the data type does not change, do not add post reorder // If mean value is not specified and the data type does not change, do not add post reorder
if (network_input_data_type != networkInputLayout.data_type) { if (network_input_data_type != networkInputLayout.data_type || connected_to_quantize(op)) {
p.add_primitive(*op, cldnn::reorder(preprocessPrimID, p.add_primitive(*op, cldnn::reorder(preprocessPrimID,
cldnn::input_info(inputName), cldnn::input_info(inputName),
networkInputLayout, networkInputLayout,