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

[CPU] Fix crash issue: RuntimeError: Primitive descriptor was not found for… (#16186)

Browse Source
This commit is contained in:
Xiping Yan 2023-03-16 14:17:06 +08:00 committed by GitHub
parent 2f3ae4518e
commit 2194552dc5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/plugins/intel_cpu/src/nodes/matmul.cpp
View File

@ -624,6 +624,7 @@ void MatMul::prepareParams() {
primitive_desc_iterator itpd = desc.createPrimitiveDescriptorIterator(engine, key.attr);
matmul::primitive_desc prim_desc;
auto itpd_first = itpd;
while (static_cast<bool>(itpd)) {
impl_desc_type impl_type = parse_impl_name(itpd.impl_info_str());
@ -631,8 +632,14 @@ void MatMul::prepareParams() {
prim_desc = itpd.get();
break;
}
if (!itpd.next_impl())
return matmul();
if (!itpd.next_impl()) {
// In case of dynamic shapes an implementation type chosen as optimal for a primitive_desc with
// undefined input shapes, is not necessarily available for the primitive_desc with defined shape.
// Example: brgemm_avx512_amx (Intel Sapphire Rapids Platform) is available for a primitive with
// undefined input shapes but not available for primitive_desc with input batch 1.
prim_desc = itpd_first.get();
break;
}
}
return matmul(prim_desc);
};

7
src/plugins/intel_cpu/tests/functional/single_layer_tests/matmul.cpp
View File

@ -1277,6 +1277,13 @@ INSTANTIATE_TEST_SUITE_P(nightly_MM_Brgemm_Amx_Static, MatMulLayerCPUTest, testB
const std::vector<ShapeRelatedParams> IS_Brgemm_Dynamic = {
{
{
{{-1, 256}, {{1, 256}}},
{{256, 384}, {{256, 384}}}
},
{false, false}
},
{
{
{{-1, -1}, {{55, 12}, {33, 7}}},