| // Package cpuinfo gives runtime info about the current CPU. | |
| // | |
| // This is a very limited module meant for use internally | |
| // in this project. For more versatile solution check | |
| // https://github.com/klauspost/cpuid. | |
| package cpuinfo | |
| // HasBMI1 checks whether an x86 CPU supports the BMI1 extension. | |
| func HasBMI1() bool { | |
| return hasBMI1 | |
| } | |
| // HasBMI2 checks whether an x86 CPU supports the BMI2 extension. | |
| func HasBMI2() bool { | |
| return hasBMI2 | |
| } | |
| // DisableBMI2 will disable BMI2, for testing purposes. | |
| // Call returned function to restore previous state. | |
| func DisableBMI2() func() { | |
| old := hasBMI2 | |
| hasBMI2 = false | |
| return func() { | |
| hasBMI2 = old | |
| } | |
| } | |
| // HasBMI checks whether an x86 CPU supports both BMI1 and BMI2 extensions. | |
| func HasBMI() bool { | |
| return HasBMI1() && HasBMI2() | |
| } | |
| var hasBMI1 bool | |
| var hasBMI2 bool |