Check for BMI instructions

SChernykh · May 30, 2024 · 7ee603a · 7ee603a
1 parent 121f6dd
commit 7ee603a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/cpu.cpp b/src/cpu.cpp
@@ -48,7 +48,7 @@ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 namespace randomx {
 
-	Cpu::Cpu() : aes_(false), ssse3_(false), avx2_(false) {
+	Cpu::Cpu() : aes_(false), ssse3_(false), bmi_(false), avx2_(false) {
 #ifdef HAVE_CPUID
 		int info[4];
 		cpuid(info, 0);
@@ -67,6 +67,7 @@ namespace randomx {
 		}
 		if (nIds >= 0x00000007) {
 			cpuid(info, 0x00000007);
+			bmi_ = (info[1] & (1 << 3)) != 0;
 			avx2_ = (info[1] & (1 << 5)) != 0;
 		}
 #elif defined(__aarch64__)

diff --git a/src/cpu.hpp b/src/cpu.hpp
@@ -39,6 +39,9 @@ namespace randomx {
 		bool hasSsse3() const {
 			return ssse3_;
 		}
+		bool hasBmi() const {
+			return bmi_;
+		}
 		bool hasAvx2() const {
 			return avx2_;
 		}
@@ -63,7 +66,7 @@ namespace randomx {
 		}
 
 	private:
-		bool aes_, ssse3_, avx2_;
+		bool aes_, ssse3_, bmi_, avx2_;
 		int manufacturer_string[4];
 		union
 		{