m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix depthwise bug

Browse Source
pull/10353/head
lzk 4 years ago
parent 1cd66fb729
commit 459762e590
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File

12
mindspore/lite/nnacl/x86_64_sse/ConvDwFp32IndirectRow.c
Unescape View File

@ -49,7 +49,7 @@ void ConvDwFp32Avx5x5(float *output, float **input, const float *weights, const
__m256 in4 = _mm256_loadu_ps(in[k + 3]); __m256 in4 = _mm256_loadu_ps(in[k + 3]);
__m256 w4 = _mm256_loadu_ps(w + 24); __m256 w4 = _mm256_loadu_ps(w + 24);
out1 = _mm256_fmadd_ps(in3, w3, out1); out1 = _mm256_fmadd_ps(in3, w3, out1);
__m256 in5 = _mm256_loadu_ps(in[k + 8]); __m256 in5 = _mm256_loadu_ps(in[k + 4]);
__m256 w5 = _mm256_loadu_ps(w + 32); __m256 w5 = _mm256_loadu_ps(w + 32);
out1 = _mm256_fmadd_ps(in4, w4, out1); out1 = _mm256_fmadd_ps(in4, w4, out1);
w += 40; w += 40;
@ -68,7 +68,10 @@ void ConvDwFp32Avx5x5(float *output, float **input, const float *weights, const
__m256 zero = _mm256_setzero_ps(); __m256 zero = _mm256_setzero_ps();
out1 = _mm256_max_ps(out1, zero); out1 = _mm256_max_ps(out1, zero);
} }
if (c == C8NUM) { if (c > C8NUM || c8_mod == 0) {
_mm256_storeu_ps(output, out1);
output += C8NUM;
} else {
__m128 tmp; __m128 tmp;
switch (c8_mod) { switch (c8_mod) {
case 1: case 1:
@ -105,10 +108,7 @@ void ConvDwFp32Avx5x5(float *output, float **input, const float *weights, const
_mm256_storeu_ps(output, out1); _mm256_storeu_ps(output, out1);
break; break;
} }
output += c8_mod == 0 ? C8NUM : c8_mod; output += c8_mod;
} else {
_mm256_storeu_ps(output, out1);
output += C8NUM;
} }
} }
} }

Write Preview
Loading…
Cancel
Save