From c074cdb1b035b0bcd0be9aa25feab97d2613140f Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Fri, 13 Jan 2023 18:44:00 +0000 Subject: IVGCVSW-6493 Bug Fix on RHS permute GpuAcc Batch MatMul workload Fp32 Signed-off-by: Teresa Charlin Change-Id: I60e9284b90467f58e0acd74d3f1493546b6f1b9b --- src/backends/cl/workloads/ClBatchMatMulWorkload.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/backends/cl/workloads/ClBatchMatMulWorkload.cpp b/src/backends/cl/workloads/ClBatchMatMulWorkload.cpp index 4acdef5e5c..ece87c2672 100644 --- a/src/backends/cl/workloads/ClBatchMatMulWorkload.cpp +++ b/src/backends/cl/workloads/ClBatchMatMulWorkload.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2022 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2022-2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -158,13 +158,13 @@ ClBatchMatMulWorkload::ClBatchMatMulWorkload(const BatchMatMulQueueDescriptor& d if (descriptor.m_Parameters.m_TransposeY == true) { armnn::PermutationVector permutationYVector - = GeneratePermutationVectorOnLastTwoDimensions(info.m_InputTensorInfos[0].GetNumDimensions()); - const TensorInfo permutedYInfo = armnnUtils::Permuted(info.m_InputTensorInfos[0], permutationYVector); + = GeneratePermutationVectorOnLastTwoDimensions(info.m_InputTensorInfos[1].GetNumDimensions()); + const TensorInfo permutedYInfo = armnnUtils::Permuted(info.m_InputTensorInfos[1], permutationYVector); const auto aclPermutationYVector = armcomputetensorutils::BuildArmComputePermutationVector(permutationYVector); armcomputetensorutils::BuildArmComputeTensor(m_PermutedTensorY, permutedYInfo); armcomputetensorutils::InitialiseArmComputeTensorEmpty(m_PermutedTensorY); - std::unique_ptr permuteLayerY(new arm_compute::CLPermute()); + auto permuteLayerY = std::make_unique(); permuteLayerY->configure(clCompileContext, &inputY, &m_PermutedTensorY, -- cgit v1.2.1