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

[CPU] Fixed segfault in Generic node creation for dynamic models (#14420)

Browse Source
This commit is contained in:
Alexandra Sidorova 2022-12-08 17:13:26 +04:00 committed by GitHub
parent 6998321f5b
commit b1700d97f1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 27 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/plugins/intel_cpu/src/nodes/generic.cpp
View File

@ -17,8 +17,34 @@ namespace ov {
namespace intel_cpu {
namespace node {
namespace {
/**
* Dummy Shape Inference while Generic op doesn't support Dynamism
*
*/
class GenericShapeInfer : public ShapeInferEmptyPads {
public:
GenericShapeInfer() = default;
std::vector<VectorDims> infer(
const std::vector<std::reference_wrapper<const VectorDims>>& input_shapes,
const std::unordered_map<size_t, MemoryPtr>& data_dependency) override {
IE_THROW(Unexpected) << "Generic operations doesn't support shape inference.";
return {};
}
port_mask_t get_port_mask() const override { return EMPTY_PORT_MASK; }
};
class GenericShapeInferFactory : public ShapeInferFactory {
public:
ShapeInferPtr makeShapeInfer() const override {
return std::make_shared<GenericShapeInfer>();
}
};
} // namespace
Generic::Generic(const std::shared_ptr<ngraph::Node>& op, const dnnl::engine& eng, WeightsSharing::Ptr &cache)
: Node(op, eng, cache, NgraphShapeInferFactory(op, EMPTY_PORT_MASK)), ngraphOp(op) {
: Node(op, eng, cache, GenericShapeInferFactory()), ngraphOp(op) {
}
void Generic::getSupportedDescriptors() {