Skip to content

Commit

Permalink
update nhwc check 2
Browse files Browse the repository at this point in the history
  • Loading branch information
allnes committed Jan 22, 2024
1 parent 972726f commit a9d9ca2
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions src/plugins/intel_cpu/src/nodes/deconv.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -474,7 +474,7 @@ void Deconvolution::getSupportedDescriptors() {
config.outConfs.resize(getOriginalOutputsNumber());

auto& creatorsMap = BlockedDescCreator::getCommonCreators();
auto checkDesc = [&](LayoutType format, LayoutType weights_format) -> bool {
auto checkDesc = [&](LayoutType format, LayoutType weights_format = LayoutType::ncsp) -> bool {
NodeConfig config;
config.inConfs.resize(getParentEdges().size());
config.outConfs.resize(getOriginalOutputsNumber());
Expand Down Expand Up @@ -506,7 +506,7 @@ void Deconvolution::getSupportedDescriptors() {

return AclDeconvExecutorBuilder::customIsSupported(deconvAttrs, srcMemoryDescs, dstMemoryDescs);
};
useACL = checkDesc(LayoutType::nspc, LayoutType::ncsp) || checkDesc(LayoutType::ncsp, LayoutType::ncsp);
useACL = checkDesc(LayoutType::nspc) || checkDesc(LayoutType::ncsp);
if (useACL) return;
#endif

Expand Down Expand Up @@ -1224,7 +1224,7 @@ void Deconvolution::initSupportedPrimitiveDescriptors() {
initPaddingR(tmpInShape, tmpOutShape);

auto& creatorsMap = BlockedDescCreator::getCommonCreators();
auto pushDesc = [&](LayoutType format, LayoutType weights_format) {
auto pushDesc = [&](LayoutType format, LayoutType weights_format = LayoutType::ncsp) {
NodeConfig config;
config.inConfs.resize(getParentEdges().size());
config.outConfs.resize(getOriginalOutputsNumber());
Expand Down Expand Up @@ -1260,8 +1260,8 @@ void Deconvolution::initSupportedPrimitiveDescriptors() {

supportedPrimitiveDescriptors.emplace_back(config, impl_desc_type::gemm_acl, factory);
};
pushDesc(LayoutType::nspc, LayoutType::ncsp);
pushDesc(LayoutType::ncsp, LayoutType::ncsp);
pushDesc(LayoutType::nspc);
pushDesc(LayoutType::ncsp);
}


Expand Down

0 comments on commit a9d9ca2

Please sign in to comment.