Searched refs:vec_x (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/mat/impls/sell/seq/ |
| H A D | sell.c | 29 #define AVX512_Mult_Private(vec_idx, vec_x, vec_vals, vec_y) \ argument
33 vec_x = _mm512_i32gather_pd(vec_idx, x, _MM_SCALE_8); \
34 vec_y = _mm512_fmadd_pd(vec_x, vec_vals, vec_y)
36 #define AVX2_Mult_Private(vec_idx, vec_x, vec_vals, vec_y) \ argument
39 vec_x = _mm256_i32gather_pd(x, vec_idx, _MM_SCALE_8); \
40 vec_y = _mm256_fmadd_pd(vec_x, vec_vals, vec_y)
330 __m512d vec_x, vec_y, vec_vals; in MatMult_SeqSELL() local
337 __m256d vec_x, vec_y, vec_y2, vec_vals; in MatMult_SeqSELL() local
342 __m256d vec_x, vec_y, vec_y2, vec_vals; in MatMult_SeqSELL() local
371 AVX512_Mult_Private(vec_idx, vec_x, vec_vals, vec_y); in MatMult_SeqSELL()
[all …]
|
| /petsc/src/mat/impls/aij/seq/aijperm/ |
| H A D | aijperm.c | 285 __m512d vec_x, vec_y, vec_vals; in MatMult_SeqAIJPERM() local
383 vec_x = _mm512_i32gather_pd(vec_idx, x, _MM_SCALE_8); in MatMult_SeqAIJPERM()
384 vec_y = _mm512_fmadd_pd(vec_x, vec_vals, vec_y); in MatMult_SeqAIJPERM()
391 vec_x = _mm512_mask_i32gather_pd(vec_x, mask, vec_idx, x, _MM_SCALE_8); in MatMult_SeqAIJPERM()
392 vec_y = _mm512_mask3_fmadd_pd(vec_x, vec_vals, vec_y, mask); in MatMult_SeqAIJPERM()
420 vec_x = _mm512_i32gather_pd(vec_idx, x, _MM_SCALE_8); in MatMult_SeqAIJPERM()
421 vec_y = _mm512_fmadd_pd(vec_x, vec_vals, vec_y); in MatMult_SeqAIJPERM()
|
| /petsc/src/mat/impls/aij/seq/ |
| H A D | aij.h | 626 __m512d vec_x, vec_y, vec_vals; in PetscSparseDensePlusDot_AVX512_Private() local
634 vec_x = _mm512_i32gather_pd(vec_idx, x, _MM_SCALE_8); in PetscSparseDensePlusDot_AVX512_Private()
635 vec_y = _mm512_fmadd_pd(vec_x, vec_vals, vec_y); in PetscSparseDensePlusDot_AVX512_Private()
646 vec_x = _mm512_mask_i32gather_pd(vec_x, mask, vec_idx, x, _MM_SCALE_8); in PetscSparseDensePlusDot_AVX512_Private()
647 vec_y = _mm512_mask3_fmadd_pd(vec_x, vec_vals, vec_y, mask); in PetscSparseDensePlusDot_AVX512_Private()
|