Quelle  cpufeature-macros.h   Sprache: C

/* SPDX-License-Identifier: GPL-2.0-only */
/*
 * Copyright 2022-2024 Rivos, Inc
 */

#ifndef _ASM_CPUFEATURE_MACROS_H
#define _ASM_CPUFEATURE_MACROS_H

#include <asm/hwcap.h>
#include <asm/alternative-macros.h>

#define STANDARD_EXT  0

bool __riscv_isa_extension_available(const unsigned long *isa_bitmap, unsigned int bit);
#define riscv_isa_extension_available(isa_bitmap, ext) \
 __riscv_isa_extension_available(isa_bitmap, RISCV_ISA_EXT_##ext)

static __always_inline bool __riscv_has_extension_likely(const unsigned long vendor,
        const unsigned long ext)
{
 asm goto(ALTERNATIVE("j %l[l_no]", "nop", %[vendor], %[ext], 1)
 :
 : [vendor] "i" (vendor), [ext] "i" (ext)
 :
 : l_no);

 return true;
l_no:
 return false;
}

static __always_inline bool __riscv_has_extension_unlikely(const unsigned long vendor,
          const unsigned long ext)
{
 asm goto(ALTERNATIVE("nop", "j %l[l_yes]", %[vendor], %[ext], 1)
 :
 : [vendor] "i" (vendor), [ext] "i" (ext)
 :
 : l_yes);

 return false;
l_yes:
 return true;
}

static __always_inline bool riscv_has_extension_unlikely(const unsigned long ext)
{
 compiletime_assert(ext < RISCV_ISA_EXT_MAX, "ext must be < RISCV_ISA_EXT_MAX");

 if (IS_ENABLED(CONFIG_RISCV_ALTERNATIVE))
  return __riscv_has_extension_unlikely(STANDARD_EXT, ext);

 return __riscv_isa_extension_available(NULL, ext);
}

static __always_inline bool riscv_has_extension_likely(const unsigned long ext)
{
 compiletime_assert(ext < RISCV_ISA_EXT_MAX, "ext must be < RISCV_ISA_EXT_MAX");

 if (IS_ENABLED(CONFIG_RISCV_ALTERNATIVE))
  return __riscv_has_extension_likely(STANDARD_EXT, ext);

 return __riscv_isa_extension_available(NULL, ext);
}

#endif /* _ASM_CPUFEATURE_MACROS_H */