diff options
Diffstat (limited to 'arm_compute/runtime/NEON/AssemblyHelper.h')
| -rw-r--r-- | arm_compute/runtime/NEON/AssemblyHelper.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/arm_compute/runtime/NEON/AssemblyHelper.h b/arm_compute/runtime/NEON/AssemblyHelper.h index ee09ef531e..39c20fab00 100644 --- a/arm_compute/runtime/NEON/AssemblyHelper.h +++ b/arm_compute/runtime/NEON/AssemblyHelper.h @@ -100,6 +100,7 @@ public: { ARM_COMPUTE_ERROR_ON(_pretranspose == nullptr || _pretranspose->buffer() == nullptr); _gemm_kernel_asm->pretranspose_B_array(reinterpret_cast<void *>(_pretranspose->buffer()), in1_ptr, ldb, multi_stride_b); + _b->mark_as_unused(); } NEScheduler::get().schedule(_optimised_kernel.get(), Window::DimX); |