diff --git a/src/operators/math/math_function.h b/src/operators/math/math_function.h index 0f281e713426beeef94a86c9c82105879b07cc80..e3184e33bd4df85b91b4ef40390f7ce2ca77fb2e 100644 --- a/src/operators/math/math_function.h +++ b/src/operators/math/math_function.h @@ -12,7 +12,7 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. */ -#pragma oncenki +#pragma once #include #include "framework/tensor.h" @@ -24,7 +24,7 @@ namespace math { template void matmul(const framework::Tensor &matrix_a, bool trans_a, const framework::Tensor &matrix_b, bool trans_b, T alpha, - framework::Tensor *matrix_out, T beta, bool relu, float *bias); + framework::Tensor *matrix_out, T beta, bool relu = false, float *bias = nullptr); template void matmulWithBn(const framework::Tensor &matrix_a, bool trans_a, diff --git a/test/common/test_gemm_perf.cpp b/test/common/test_gemm_perf.cpp index 9b62ffe9e17790f068caaa94e9e82c53e5c2d9fb..c505c61fce21775136a368949a451999b97b3069 100644 --- a/test/common/test_gemm_perf.cpp +++ b/test/common/test_gemm_perf.cpp @@ -51,7 +51,7 @@ int main() { for (int j = 0; j < 10; ++j) { paddle_mobile::operators::math::matmul( aa, false, bb, false, static_cast(1), &cc, static_cast(0), - false, ccptr); + false, biasptr); // paddle_mobile::operators::math::matmulWithBn( // aa, false, bb, false, static_cast(1), &cc,