1 /* 2 * Copyright (c) 2020 Loongson Technology Corporation Limited 3 * Contributed by Shiyou Yin <yinshiyou-hf@loongson.cn> 4 * 5 * This file is part of FFmpeg. 6 * 7 * FFmpeg is free software; you can redistribute it and/or 8 * modify it under the terms of the GNU Lesser General Public 9 * License as published by the Free Software Foundation; either 10 * version 2.1 of the License, or (at your option) any later version. 11 * 12 * FFmpeg is distributed in the hope that it will be useful, 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 15 * Lesser General Public License for more details. 16 * 17 * You should have received a copy of the GNU Lesser General Public 18 * License along with FFmpeg; if not, write to the Free Software 19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 20 */ 21 22 #include <stdint.h> 23 #include "cpu.h" 24 25 #define LOONGARCH_CFG2 0x2 26 #define LOONGARCH_CFG2_LSX (1 << 6) 27 #define LOONGARCH_CFG2_LASX (1 << 7) 28 cpu_flags_cpucfg(void)29static int cpu_flags_cpucfg(void) 30 { 31 int flags = 0; 32 uint32_t cfg2 = 0; 33 34 __asm__ volatile( 35 "cpucfg %0, %1 \n\t" 36 : "+&r"(cfg2) 37 : "r"(LOONGARCH_CFG2) 38 ); 39 40 if (cfg2 & LOONGARCH_CFG2_LSX) 41 flags |= AV_CPU_FLAG_LSX; 42 43 if (cfg2 & LOONGARCH_CFG2_LASX) 44 flags |= AV_CPU_FLAG_LASX; 45 46 return flags; 47 } 48 ff_get_cpu_flags_loongarch(void)49int ff_get_cpu_flags_loongarch(void) 50 { 51 #if defined __linux__ 52 return cpu_flags_cpucfg(); 53 #else 54 /* Assume no SIMD ASE supported */ 55 return 0; 56 #endif 57 } 58 ff_get_cpu_max_align_loongarch(void)59size_t ff_get_cpu_max_align_loongarch(void) 60 { 61 int flags = av_get_cpu_flags(); 62 63 if (flags & AV_CPU_FLAG_LASX) 64 return 32; 65 if (flags & AV_CPU_FLAG_LSX) 66 return 16; 67 68 return 8; 69 } 70