Samuka007/asterinas
1
0
Fork 0
mirror of https://github.com/asterinas/asterinas.git synced 2025-06-09 13:26:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix the AVX initialization bugs on lower-end CPUs

Browse Source
This commit is contained in:
Zhe Tang 2025-04-07 13:34:19 +00:00 committed by Ruihan Li
parent b833ec6ede
commit 36f6f9bcd4
1 changed files with 20 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
ostd/src/arch/x86/mod.rs
View File

@ -163,6 +163,20 @@ pub fn read_random() -> Option<u64> {
None
}
fn has_avx() -> bool {
use core::arch::x86_64::{__cpuid, __cpuid_count};
let cpuid_result = unsafe { __cpuid(0) };
if cpuid_result.eax < 1 {
// CPUID function 1 is not supported
return false;
}
let cpuid_result = unsafe { __cpuid_count(1, 0) };
// Check for AVX (bit 28 of ecx)
cpuid_result.ecx & (1 << 28) != 0
}
fn has_avx512() -> bool {
use core::arch::x86_64::{__cpuid, __cpuid_count};
@ -198,7 +212,12 @@ pub(crate) fn enable_cpu_features() {
}
let mut xcr0 = x86_64::registers::xcontrol::XCr0::read();
xcr0 |= XCr0Flags::AVX | XCr0Flags::SSE;
xcr0 |= XCr0Flags::SSE;
if has_avx() {
xcr0 |= XCr0Flags::AVX;
}
if has_avx512() {
xcr0 |= XCr0Flags::OPMASK | XCr0Flags::ZMM_HI256 | XCr0Flags::HI16_ZMM;