Skip to content

Commit

Permalink
WIP: [CPU] Deconvolution: Added fp32 output precision check for fusing
Browse files Browse the repository at this point in the history
  • Loading branch information
antonvor committed Nov 21, 2023
1 parent 61c5a80 commit ed71c10
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/plugins/intel_cpu/src/nodes/deconv.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -329,7 +329,7 @@ bool Deconvolution::canFuse(const NodePtr& node) const {
if (canBeExecutedInInt8())
return canFuseSimpleOperation(node);

return (fusedWith.empty() && node->canBePerformedAsScaleShift(this));
return (fusedWith.empty() && node->canBePerformedAsScaleShift(this)) && node->getOriginalOutputPrecisionAtPort(0) == ov::element::f32;
}

std::pair<VectorDims, VectorDims> Deconvolution::makeDummyInOutShape() {
Expand Down

0 comments on commit ed71c10

Please sign in to comment.