refine vmul jitcode

test=develop
fix_recordio_link
tensor-tang 7 years ago
parent a764e900a5
commit 85bcb286f5

@ -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

Loading…
Cancel
Save