In place no new alloc multiply

int32_t* = reinterpret_cast<float32*> for in-place no alloc multiply.
jerinphilip · Apr 20, 2022 · 3c03bcf · 3c03bcf
1 parent d908352
commit 3c03bcf
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/MozIntGemm/moz_intgemm_ruy.inl b/MozIntGemm/moz_intgemm_ruy.inl
@@ -86,9 +86,7 @@ void int8MultiplyAndAddBias(const int8_t *input_A_prepared, float scale_A,
   ruy::MakeSimpleLayout(rows_A, cols_B, ruy::Order::kRowMajor,
                         dst.mutable_layout());
 
-  detail::AlignedVector<std::int32_t> dst_data(rows_A * cols_B);
-  std::int32_t *dest_ptr = dst_data.data();
-
+  std::int32_t *dest_ptr = reinterpret_cast<std::int32_t *>(output);
   dst.set_data(dest_ptr);
 
   // When Dst is int32, mul_params is unused.