wip

Tencent · Sep 20, 2024 · ecf2e3b · ecf2e3b
1 parent 47cd674
commit ecf2e3b
Show file tree

Hide file tree

Showing 2 changed files with 1,315 additions and 342 deletions.
diff --git a/src/layer/arm/gemm_arm.cpp b/src/layer/arm/gemm_arm.cpp
@@ -6160,21 +6160,6 @@ int Gemm_arm::forward_int8(const std::vector<Mat>& bottom_blobs, std::vector<Mat
  // 1xN
  broadcast_type_C = 4;
  }
-
- // pre-multiply C with beta
- if (beta != 1.f)
- {
- Mat CT_data;
- CT_data.create_like(C, opt.workspace_allocator);
-
- const int size = C.total() * C.elempack;
- for (int i = 0; i < size; i++)
- {
- CT_data[i] = C[i] * beta;
- }
-
- C = CT_data;
- }
  }
  }
 
@@ -6242,22 +6227,8 @@ int Gemm_arm::forward_int8(const std::vector<Mat>& bottom_blobs, std::vector<Mat
  const Mat& B = bottom_blobs[1];
  ret = gemm_arm_int8(A, B, C, top_blob, broadcast_type_C, transA, transB, output_transpose, constant_TILE_M, constant_TILE_N, constant_TILE_K, _nT, opt);
  }
- if (ret != 0)
- return ret;
-
- // multiply top_blob with alpha
- if (alpha != 1.f)
- {
- const int size = top_blob.total() * out_elempack;
 
- #pragma omp parallel for num_threads(opt.num_threads)
- for (int i = 0; i < size; i++)
- {
- top_blob[i] *= alpha;
- }
- }
-
- return 0;
+ return ret;
 }
 #endif