diff --git a/libavutil/riscv/bswap.h b/libavutil/riscv/bswap.h index ce75de974e..886893e241 100644 --- a/libavutil/riscv/bswap.h +++ b/libavutil/riscv/bswap.h @@ -22,11 +22,51 @@ #include #include "config.h" #include "libavutil/attributes.h" +#include "libavutil/riscv/cpu.h" #if defined (__GNUC__) || defined (__clang__) #define av_bswap16 __builtin_bswap16 -#define av_bswap32 __builtin_bswap32 -#define av_bswap64 __builtin_bswap64 + +static av_always_inline av_const uint32_t av_bswap32_rv(uint32_t x) +{ +#if HAVE_RV && !defined(__riscv_zbb) + if (!__builtin_constant_p(x) && + __builtin_expect(ff_rv_zbb_support(), 1)) { + uintptr_t y; + + __asm__ ( + ".option push\n" + ".option arch, +zbb\n" + "rev8 %0, %1\n" + ".option pop" : "=r" (y) : "r" (x)); + return y >> (__riscv_xlen - 32); + } +#endif + return __builtin_bswap32(x); +} +#define av_bswap32 av_bswap32_rv + +#if __riscv_xlen >= 64 +static av_always_inline av_const uint64_t av_bswap64_rv(uint64_t x) +{ +#if HAVE_RV && !defined(__riscv_zbb) + if (!__builtin_constant_p(x) && + __builtin_expect(ff_rv_zbb_support(), 1)) { + uintptr_t y; + + __asm__ ( + ".option push\n" + ".option arch, +zbb\n" + "rev8 %0, %1\n" + ".option pop" : "=r" (y) : "r" (x)); + return y >> (__riscv_xlen - 64); + } +#endif + return __builtin_bswap64(x); +} +#define av_bswap64 av_bswap64_rv +#endif + #endif #endif /* AVUTIL_RISCV_BSWAP_H */