diff --git a/libavfilter/x86/scene_sad.asm b/libavfilter/x86/scene_sad.asm index bf7236b3a3..2cd9dddb5c 100644 --- a/libavfilter/x86/scene_sad.asm +++ b/libavfilter/x86/scene_sad.asm @@ -72,3 +72,10 @@ INIT_YMM avx2 SAD_FRAMES %endif + +%if HAVE_AVX512_EXTERNAL + +INIT_ZMM avx512 +SAD_FRAMES + +%endif diff --git a/libavfilter/x86/scene_sad_init.c b/libavfilter/x86/scene_sad_init.c index 4a4c40195f..2d631b376a 100644 --- a/libavfilter/x86/scene_sad_init.c +++ b/libavfilter/x86/scene_sad_init.c @@ -41,6 +41,9 @@ SCENE_SAD_FUNC(scene_sad_sse2, ff_scene_sad_sse2, 16) #if HAVE_AVX2_EXTERNAL SCENE_SAD_FUNC(scene_sad_avx2, ff_scene_sad_avx2, 32) #endif +#if HAVE_AVX512_EXTERNAL +SCENE_SAD_FUNC(scene_sad_avx512, ff_scene_sad_avx512, 64) +#endif #endif ff_scene_sad_fn ff_scene_sad_get_fn_x86(int depth) @@ -48,6 +51,10 @@ ff_scene_sad_fn ff_scene_sad_get_fn_x86(int depth) #if HAVE_X86ASM int cpu_flags = av_get_cpu_flags(); if (depth <= 8) { +#if HAVE_AVX512_EXTERNAL + if (EXTERNAL_AVX512(cpu_flags)) + return scene_sad_avx512; +#endif #if HAVE_AVX2_EXTERNAL if (EXTERNAL_AVX2_FAST(cpu_flags)) return scene_sad_avx2;