language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-27 05:49:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

prefix assembly functions with _ on macos

Browse source
This commit is contained in:
Brendan Hansknecht 2023-06-02 15:10:14 -07:00
parent 60babcc56e
commit 341ef9bb7b
No known key found for this signature in database
GPG key ID: 0EA784685083E75B
7 changed files with 84 additions and 72 deletions
Download patch file Download diff file Expand all files Collapse all files

114
crates/compiler/builtins/bitcode/src/libc/cpuid.zig
View file

@ -1,67 +1,69 @@
const std = @import("std");
const builtin = @import("builtin");
const arch = builtin.cpu.arch;
const function_prefix = @import("assembly_util.zig").function_prefix;
// I couldn't manage to define this in a PIE friendly way with inline assembly.
// Instead, I am defining it as global assembly functions.
comptime {
switch (arch) {
.x86_64 => {
asm (
\\ # Check if AVX2 is supported.
\\ # Returns 1 if AVX2 is supported, 0 otherwise.
\\ .global supports_avx2;
\\ supports_avx2:
\\ # Save the EBX register.
\\ push %rbx
\\
\\ # Call the CPUID instruction with the EAX register set to 7 and ECX set to 0.
\\ # This will get the CPUID information for the current CPU.
\\ mov $7, %eax
\\ mov $0, %ecx
\\ cpuid
\\
\\ # The AVX2 feature flag is located in the EBX register at bit 5.
\\ bt $5, %ebx
\\ jc .avx2_supported
\\
\\ # AVX2 is not supported.
\\ pop %rbx
\\ mov $0, %eax
\\ ret
\\
\\ .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
);
asm (std.fmt.comptimePrint(
\\ # Check if AVX2 is supported.
\\ # Returns 1 if AVX2 is supported, 0 otherwise.
\\ .global {[function_prefix]s}supports_avx2;
\\ {[function_prefix]s}supports_avx2:
\\ # Save the EBX register.
\\ push %rbx
\\
\\ # Call the CPUID instruction with the EAX register set to 7 and ECX set to 0.
\\ # This will get the CPUID information for the current CPU.
\\ mov $7, %eax
\\ mov $0, %ecx
\\ cpuid
\\
\\ # The AVX2 feature flag is located in the EBX register at bit 5.
\\ bt $5, %ebx
\\ jc .avx2_supported
\\
\\ # AVX2 is not supported.
\\ pop %rbx
\\ mov $0, %eax
\\ ret
\\
\\ .avx2_supported:
\\ pop %rbx
\\ mov $1, %eax
\\ ret
, .{ .function_prefix = function_prefix }));
asm (std.fmt.comptimePrint(
\\ # Check if prefetchw is supported.
\\ # Returns 1 if the prefetchw instruction is supported, 0 otherwise.
\\ .global {[function_prefix]s}supports_prefetchw;
\\ {[function_prefix]s}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
, .{ .function_prefix = function_prefix }));
},
else => unreachable,
}