[ hgemm ] Use aligned memory allocation in transpose / padding gemm

- Using unaligned memory may invoke SIGSEGV **Self evaluation:** 1. Build test: [X]Passed [ ]Failed [ ]Skipped 2. Run test: [X]Passed [ ]Failed [ ]Skipped Signed-off-by: skykongkong8 <[email protected]>
nnstreamer · Jul 2, 2024 · 2ff7d98 · 2ff7d98
1 parent dc38b83
commit 2ff7d98
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/nntrainer/tensor/hgemm/hgemm.cpp b/nntrainer/tensor/hgemm/hgemm.cpp
@@ -1338,7 +1338,6 @@ void hgemm_transB(const __fp16 *A, const __fp16 *B, float *C, unsigned int M,
 void hgemm_transA(const __fp16 *A, const __fp16 *B, float *C, unsigned int M,
                   unsigned int N, unsigned int K, float alpha, float beta) {
   __fp16 *A_T = alignedMalloc(M * K);
-
   transpose_neon<__fp16>(K, M, A, M, A_T, K);
 
   hgemm_noTrans(A_T, B, C, M, N, K, alpha, beta);