better extract musl lib

This commit is contained in:
Brendan Hansknecht 2023-06-01 21:09:28 -07:00
parent 159f95aad1
commit 77624f627b
No known key found for this signature in database
GPG key ID: 0EA784685083E75B
5 changed files with 65 additions and 31 deletions

View file

@ -2,7 +2,7 @@ const builtin = @import("builtin");
const arch = builtin.cpu.arch;
// I couldn't manage to define this in a PIE friendly way with inline assembly.
// Instead, I am definining it as a global assembly function.
// Instead, I am definining it as global assembly functions.
comptime {
switch (arch) {
.x86_64 => {
@ -20,17 +20,44 @@ comptime {
\\ mov $0, %ecx
\\ cpuid
\\
\\ # Check if the AVX2 feature flag is set.
\\ # The AVX2 feature flag is located in the EBX register at bit 5.
\\ bt $5, %ebx
\\ jc .AVX2Supported
\\ jc .avx2_supported
\\
\\ # AVX2 is not supported.
\\ pop %rbx
\\ mov $0, %eax
\\ ret
\\
\\ .AVX2Supported:
\\ .avx2_supported:
\\ pop %rbx
\\ mov $1, %eax
\\ ret
);
asm (
\\ # Check if prefetchw is supported.
\\ # Returns 1 if the prefetchw instruction is supported, 0 otherwise.
\\ .global supports_prefetchw;
\\ supports_prefetchw:
\\ # Save the EBX register.
\\ push %rbx
\\
\\ # Call the CPUID instruction with the EAX register set to 0x80000001 and ECX set to 0.
\\ # This will get the CPUID information for the current CPU.
\\ mov $0x80000001, %eax
\\ mov $0, %ecx
\\ cpuid
\\
\\ # The prefetchw feature flag is located in the ECX register at bit 8.
\\ bt $8, %ecx
\\ jc .prefetchw_supported
\\
\\ # AVX2 is not supported.
\\ pop %rbx
\\ mov $0, %eax
\\ ret
\\
\\ .prefetchw_supported:
\\ pop %rbx
\\ mov $1, %eax
\\ ret
@ -41,3 +68,4 @@ comptime {
}
pub extern fn supports_avx2() bool;
pub extern fn supports_prefetchw() bool;