mirror of
https://github.com/roc-lang/roc.git
synced 2025-09-27 05:49:08 +00:00
better extract musl lib
This commit is contained in:
parent
159f95aad1
commit
77624f627b
5 changed files with 65 additions and 31 deletions
|
@ -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;
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue