diff options
Diffstat (limited to 'std')
| -rw-r--r-- | std/math/sqrt.zig | 18 | ||||
| -rw-r--r-- | std/math/x86_64/sqrt.zig | 15 |
2 files changed, 29 insertions, 4 deletions
diff --git a/std/math/sqrt.zig b/std/math/sqrt.zig index 4d84756b4a..18cee6e248 100644 --- a/std/math/sqrt.zig +++ b/std/math/sqrt.zig @@ -18,11 +18,21 @@ pub fn sqrt(x: var) -> (if (@typeId(@typeOf(x)) == TypeId.Int) @IntType(false, @ return T(sqrt64(x)); }, TypeId.Float => { - return switch (T) { - f32 => sqrt32(x), - f64 => sqrt64(x), + switch (T) { + f32 => { + switch (builtin.arch) { + builtin.Arch.x86_64 => return @import("x86_64/sqrt.zig").sqrt32(x), + else => return sqrt32(x), + } + }, + f64 => { + switch (builtin.arch) { + builtin.Arch.x86_64 => return @import("x86_64/sqrt.zig").sqrt64(x), + else => return sqrt64(x), + } + }, else => @compileError("sqrt not implemented for " ++ @typeName(T)), - }; + } }, TypeId.IntLiteral => comptime { if (x > @maxValue(u128)) { diff --git a/std/math/x86_64/sqrt.zig b/std/math/x86_64/sqrt.zig new file mode 100644 index 0000000000..fbb898c297 --- /dev/null +++ b/std/math/x86_64/sqrt.zig @@ -0,0 +1,15 @@ +pub fn sqrt32(x: f32) -> f32 { + return asm ( + \\sqrtss %%xmm0, %%xmm0 + : [ret] "={xmm0}" (-> f32) + : [x] "{xmm0}" (x) + ); +} + +pub fn sqrt64(x: f64) -> f64 { + return asm ( + \\sqrtsd %%xmm0, %%xmm0 + : [ret] "={xmm0}" (-> f64) + : [x] "{xmm0}" (x) + ); +} |
