aboutsummaryrefslogtreecommitdiff
path: root/std/math
diff options
context:
space:
mode:
Diffstat (limited to 'std/math')
-rw-r--r--std/math/sqrt.zig18
-rw-r--r--std/math/x86_64/sqrt.zig15
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)
+ );
+}