float_dsp_init.c 1.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. /*
  2. * This file is part of FFmpeg.
  3. *
  4. * FFmpeg is free software; you can redistribute it and/or
  5. * modify it under the terms of the GNU Lesser General Public
  6. * License as published by the Free Software Foundation; either
  7. * version 2.1 of the License, or (at your option) any later version.
  8. *
  9. * FFmpeg is distributed in the hope that it will be useful,
  10. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  12. * Lesser General Public License for more details.
  13. *
  14. * You should have received a copy of the GNU Lesser General Public
  15. * License along with FFmpeg; if not, write to the Free Software
  16. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  17. */
  18. #include "config.h"
  19. #include "libavutil/cpu.h"
  20. #include "libavutil/float_dsp.h"
  21. #include "cpu.h"
  22. extern void ff_vector_fmul_sse(float *dst, const float *src0, const float *src1,
  23. int len);
  24. extern void ff_vector_fmul_avx(float *dst, const float *src0, const float *src1,
  25. int len);
  26. extern void ff_vector_fmac_scalar_sse(float *dst, const float *src, float mul,
  27. int len);
  28. extern void ff_vector_fmac_scalar_avx(float *dst, const float *src, float mul,
  29. int len);
  30. void ff_float_dsp_init_x86(AVFloatDSPContext *fdsp)
  31. {
  32. int mm_flags = av_get_cpu_flags();
  33. if (EXTERNAL_SSE(mm_flags)) {
  34. fdsp->vector_fmul = ff_vector_fmul_sse;
  35. fdsp->vector_fmac_scalar = ff_vector_fmac_scalar_sse;
  36. }
  37. if (EXTERNAL_AVX(mm_flags)) {
  38. fdsp->vector_fmul = ff_vector_fmul_avx;
  39. fdsp->vector_fmac_scalar = ff_vector_fmac_scalar_avx;
  40. }
  41. }