From 85bcb286f5645ad81f67a86ada916ed8d0f8931b Mon Sep 17 00:00:00 2001
From: tensor-tang <tangjian03@baidu.com>
Date: Thu, 1 Nov 2018 15:19:17 +0000
Subject: [PATCH] refine vmul jitcode

test=develop
---
 paddle/fluid/operators/math/jit_code.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddle/fluid/operators/math/jit_code.cc b/paddle/fluid/operators/math/jit_code.cc
index 29a89bca982..06cf82513df 100644
--- a/paddle/fluid/operators/math/jit_code.cc
+++ b/paddle/fluid/operators/math/jit_code.cc
@@ -35,7 +35,7 @@ bool VMulJitCode::init(int d) {
 }
 
 void VMulJitCode::generate() {
-  preCode();
+  // do not need push stack, and do not need save avx512reg if do not use avx512
   int stride = sizeof(float) * AVX_FLOAT_BLOCK;
   for (int i = 0; i < num_ / AVX_FLOAT_BLOCK; ++i) {
     vmovups(ymm_src1, ptr[param1 + i * stride]);
@@ -43,7 +43,7 @@ void VMulJitCode::generate() {
     vmulps(ymm_dst, ymm_src1, ymm_src2);
     vmovups(ptr[param3 + stride * i], ymm_dst);
   }
-  postCode();
+  ret();
 }
 
 }  // namespace gen
-- 
GitLab