diff options
| author | Pablo Marquez Tello <pablo.tello@arm.com> | 2021-12-06 12:15:00 +0000 |
|---|---|---|
| committer | Pablo Marquez Tello <pablo.tello@arm.com> | 2022-05-17 22:15:15 +0000 |
| commit | 6fd75b90e736aefd837d74d2a3877009385a5546 (patch) | |
| tree | 1304ebbf150f693fe349582944c6fa8502d5175d /src/core/NEON/kernels | |
| parent | 6db4cf225a06fd42e806cf092e30eb925083ca3f (diff) | |
| download | ComputeLibrary-6fd75b90e736aefd837d74d2a3877009385a5546.tar.gz | |
DepthwiseConv reports full assembly kernel name
* Resolves MLCE-706
Change-Id: Ia15c925c13464397c79056dffe2a756e06020682
Signed-off-by: Pablo Marquez Tello <pablo.tello@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/7571
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/NEON/kernels')
| -rw-r--r-- | src/core/NEON/kernels/arm_conv/depthwise/depthwise_implementation.hpp | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/src/core/NEON/kernels/arm_conv/depthwise/depthwise_implementation.hpp b/src/core/NEON/kernels/arm_conv/depthwise/depthwise_implementation.hpp index 0665fa3a29..1ee19e5075 100644 --- a/src/core/NEON/kernels/arm_conv/depthwise/depthwise_implementation.hpp +++ b/src/core/NEON/kernels/arm_conv/depthwise/depthwise_implementation.hpp @@ -136,7 +136,14 @@ UniqueDepthwiseCommon<TInput, TWeight, TOutput> depthwise(const DepthwiseArgs &a { const DepthwiseImplementation<TInput, TWeight, TOutput, OutputStage> *impl = nullptr; const bool success = find_implementation<TInput, TWeight, TOutput, OutputStage>(args, os, impl); - return UniqueDepthwiseCommon<TInput, TWeight, TOutput>(success ? impl->get_instance(args, os) : nullptr); + + if(success) + { + auto i = impl->get_instance(args, os); + i->set_name(impl->name); + return UniqueDepthwiseCommon<TInput, TWeight, TOutput>(i); + } + return nullptr; } } // namespace depthwise |