diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-05-09 20:31:35 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:51:19 +0000 |
| commit | 28a8493b4e7617334e72e56a2151627877ef52d3 (patch) | |
| tree | 60bfb850375f430e8ff27f00f573c07c20c75620 | |
| parent | 71ac9c0e6f568231bab34bb5e19453a0d45fcff5 (diff) | |
| download | ComputeLibrary-28a8493b4e7617334e72e56a2151627877ef52d3.tar.gz | |
COMPMID-1131: Fix SIGBUS error
Forcing 128-byte alignment on pretransposed buffer.
Change-Id: I534c5ebb30ffc9a219e7a3e2d495b4b83b0d1fea
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/130614
Reviewed-by: Pablo Tello <pablo.tello@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
| -rw-r--r-- | arm_compute/runtime/NEON/AssemblyHelper.h | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/arm_compute/runtime/NEON/AssemblyHelper.h b/arm_compute/runtime/NEON/AssemblyHelper.h index 39c20fab00..3db419e148 100644 --- a/arm_compute/runtime/NEON/AssemblyHelper.h +++ b/arm_compute/runtime/NEON/AssemblyHelper.h @@ -98,8 +98,13 @@ public: _gemm_kernel_asm->set_arrays(in0_ptr, lda, batch_stride_a, multi_stride_a, in1_ptr, ldb, multi_stride_b, out_ptr, ldd, batch_stride_d, multi_stride_d); if(_gemm_kernel_asm->B_pretranspose_required()) { + // Forcing 128-byte alignment (required by 32-bit kernels) + const unsigned int alignment = 128; + void *raw_ptr = reinterpret_cast<void *>(_pretranspose->buffer()); + size_t space = _pretranspose->info()->total_size(); + void *aligned_ptr = support::cpp11::align(alignment, _gemm_kernel_asm->get_B_pretransposed_array_size(), raw_ptr, space); 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); + _gemm_kernel_asm->pretranspose_B_array(aligned_ptr, in1_ptr, ldb, multi_stride_b); _b->mark_as_unused(); } @@ -190,8 +195,10 @@ inline bool setup_assembly_kernel(const ITensor *a, const ITensor *b, ITensor *d // Check for pre-transposed support if(asm_gemm->B_pretranspose_required()) { - const size_t B_pretranspose_size = asm_gemm->get_B_pretransposed_array_size(); - allocate_workspace(B_pretranspose_size, B_pretranspose, nullptr, 1, 1); + // Forcing 128-byte alignment (required by 32-bit kernels) + const unsigned int alignment = 128; + const size_t B_pretranspose_size = asm_gemm->get_B_pretransposed_array_size(); + allocate_workspace(B_pretranspose_size, B_pretranspose, nullptr, alignment, 1); ARM_COMPUTE_ERROR_ON_NULLPTR(B_pretranspose.buffer()); asm_glue._pretranspose = &B_pretranspose; } |