diff options
Diffstat (limited to 'src/backends/cl/workloads/ClGatherNdWorkload.cpp')
-rw-r--r-- | src/backends/cl/workloads/ClGatherNdWorkload.cpp | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/src/backends/cl/workloads/ClGatherNdWorkload.cpp b/src/backends/cl/workloads/ClGatherNdWorkload.cpp index 1351f9685f..4e9dd7526f 100644 --- a/src/backends/cl/workloads/ClGatherNdWorkload.cpp +++ b/src/backends/cl/workloads/ClGatherNdWorkload.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2022-2023 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2022-2024 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -142,8 +142,6 @@ ClGatherNdWorkload::ClGatherNdWorkload(const GatherNdQueueDescriptor& descriptor flattenedCoeff_Info.SetShape({ keyIndices["ND"] }); BuildArmComputeTensor(m_FlattenedCoeff, flattenedCoeff_Info); armcomputetensorutils::InitialiseArmComputeTensorEmpty(m_FlattenedCoeff); - ARMNN_ASSERT_MSG(indicesInfo.GetDataType() == DataType::Signed32, - "flattenedCoeff must be same data type as m_FlattenedCoeff"); CopyArmComputeClTensorData<int32_t>(m_FlattenedCoeff, flattenedCoeff.data()); // Prepare the tensor to store the output of the multiplication |