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

[CPU]Fix profile graph (#23699) 

### Details:
 - *Fix profile graph construction*

### Tickets:
 - *CVS-136859*
 - https://github.com/openvinotoolkit/openvino.genai/issues/327
This commit is contained in:
Zhang Yi 2024-04-22 17:09:27 +08:00 committed by GitHub
parent 586065e293
commit 147cf78079
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/plugins/intel_cpu/src/nodes/broadcast.cpp
View File

@ -103,8 +103,7 @@ void Broadcast::getSupportedDescriptors() {
void Broadcast::initSupportedPrimitiveDescriptors() {
if (!supportedPrimitiveDescriptors.empty())
return;
supportedPrimitiveDescriptors = getSupportedConfigs(this);
supportedPrimitiveDescriptors = getSupportedConfigs(this, outputShapes.size());
}
bool Broadcast::needPrepareParams() const {

4
src/plugins/intel_cpu/src/nodes/common/tile_broadcast_utils.cpp
View File

@ -90,7 +90,7 @@ bool TileBroadcastCommon::canBeExecutedInNSPCLayout(VectorDims srcBlockedDims, V
return optimizedDims.size() <= maxNDims;
}
std::vector<NodeDesc> TileBroadcastCommon::getSupportedConfigs(const Node *node) {
std::vector<NodeDesc> TileBroadcastCommon::getSupportedConfigs(const Node *node, size_t outSize) {
std::vector<NodeDesc> supportedPrimitiveDescriptors;
auto precision = node->getOriginalInputPrecisionAtPort(0);
auto dataType = DnnlExtensionUtils::ElementTypeToDataType(precision);
@ -122,7 +122,7 @@ std::vector<NodeDesc> TileBroadcastCommon::getSupportedConfigs(const Node *node)
config.inConfs[2].setMemDesc(std::make_shared<CpuBlockedMemoryDesc>(ov::element::i32, node->getInputShapeAtPort(2)));
}
config.outConfs.resize(node->getChildEdges().size());
config.outConfs.resize(outSize);
auto pushDesc = [&](dnnl::memory::format_tag inFormat, dnnl::memory::format_tag outFormat) {
config.inConfs[0].setMemDesc(std::make_shared<DnnlBlockedMemoryDesc>(node->getInputShapeAtPort(0), dataType, inFormat));

2
src/plugins/intel_cpu/src/nodes/common/tile_broadcast_utils.h
View File

@ -16,7 +16,7 @@ namespace intel_cpu {
class TileBroadcastCommon {
protected:
static VectorDims calculateDenseStrides(const VectorDims &dims);
std::vector<NodeDesc> getSupportedConfigs(const Node *node);
std::vector<NodeDesc> getSupportedConfigs(const Node *node, size_t outSize);
bool prepareOptimizedParams(const Node *node, VectorDims& srcBlockedDims, VectorDims& dstBlockedDims);
void optimizedExecute(const MemoryPtr& srcMemory, const MemoryPtr& dstMemory);

2
src/plugins/intel_cpu/src/nodes/tile.cpp
View File

@ -110,7 +110,7 @@ void Tile::initSupportedPrimitiveDescriptors() {
if (!supportedPrimitiveDescriptors.empty())
return;
supportedPrimitiveDescriptors = getSupportedConfigs(this);
supportedPrimitiveDescriptors = getSupportedConfigs(this, outputShapes.size());
}
bool Tile::needPrepareParams() const {