Check for BMI instructions

SChernykh · May 31, 2024 · 1b1f6d2 · 1b1f6d2
1 parent 121f6dd
commit 1b1f6d2
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/cpu.cpp b/src/cpu.cpp
@@ -26,6 +26,10 @@ OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
 
+#ifdef _MSC_VER
+#pragma warning(disable : 4514)
+#endif
+
 #include "cpu.hpp"
 
 #if defined(_M_X64) || defined(__x86_64__)
@@ -48,7 +52,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 +71,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
 		{