From a30d757cfd4ad217bad1f6b6bd9e405506e05fab Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ewa=20Tusie=C5=84?= Date: Mon, 7 Jun 2021 11:58:22 +0200 Subject: [PATCH] [IE CLDNN] Convert permutation order in MatMul (#6005) --- inference-engine/src/cldnn_engine/ops/matmul.cpp | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/inference-engine/src/cldnn_engine/ops/matmul.cpp b/inference-engine/src/cldnn_engine/ops/matmul.cpp index b18e199581c..5fe6f143258 100644 --- a/inference-engine/src/cldnn_engine/ops/matmul.cpp +++ b/inference-engine/src/cldnn_engine/ops/matmul.cpp @@ -83,10 +83,11 @@ void CreateMatMulOp(Program& p, const std::shared_ptr& o for (auto o = transpose_order.size(); o < 4; o++) transpose_order.push_back((uint16_t)o); + std::vector cldnn_permute_order = ConvertPermuteOrder(transpose_order); auto permuteName = op->get_friendly_name() + "/transpose_b"; auto permutePrim = cldnn::permute(permuteName, weightsName, - transpose_order); + cldnn_permute_order); p.AddPrimitive(permutePrim); p.AddInnerPrimitiveToProfiler(permuteName, layerName, op); weightsName = permuteName; @@ -102,10 +103,11 @@ void CreateMatMulOp(Program& p, const std::shared_ptr& o for (auto o = transpose_order.size(); o < 4; o++) transpose_order.push_back((uint16_t)o); + std::vector cldnn_permute_order = ConvertPermuteOrder(transpose_order); auto permuteName = op->get_friendly_name() + "/transpose_a"; auto permutePrim = cldnn::permute(permuteName, inputName, - transpose_order); + cldnn_permute_order); p.AddPrimitive(permutePrim); p.AddInnerPrimitiveToProfiler(permuteName, layerName, op); inputName = permuteName;