Merge pull request #1 from orsinium-labs/wasm-asm

Optimized code for wasm

README.md

@@ -28,16 +28,17 @@ Here is a comparison of WebAssembly binary size (built with TinyGo) when using t
 
 | function     | tinymath | stdlib | ratio |
 | ------------ | --------:| ------:| -----:|
+| atan         |      106 |    367 |   28% |
 | atan2        |      167 |    782 |   21% |
-| exp          |      535 |   2722 |   19% |
-| fract        |      206 |    154 |  133% |
-| hypot        |       94 |    203 |   46% |
+| exp          |      463 |   2722 |   17% |
+| fract        |      166 |    154 |  107% |
+| hypot        |       67 |    203 |   33% |
 | ln           |      196 |   4892 |    4% |
-| powf         |      859 |   9167 |    9% |
+| powf         |      701 |   9167 |    7% |
 | round        |      129 |    171 |   75% |
-| sin          |      198 |   1237 |   16% |
-| trunc        |      136 |     57 |  238% |
+| sin          |      125 |   1237 |   10% |
+| sqrt         |       57 |     57 |  100% |
+| tan          |      138 |   1137 |   12% |
+| trunc        |       57 |     57 |  100% |
 
 To reproduce: `python3 size_bench.py`
-
-The two functions that are bigger in tinymath (but still small!) are the ones that have an optimized wasm-specific assembly in the standard library implementation. We're working on adding such assembly code on our side as well.

arch_nowasm.go

@@ -1,3 +1,5 @@
+//go:build !tinygo.wasm
+
 package tinymath
 
 // Functions that can be optimized for wasm
@@ -16,6 +18,17 @@ func Floor(self float32) float32 {
 	return float32(res)
 }
 
+// Approximates the square root of a number with an average deviation of ~5%.
+//
+// Returns [`NAN`] if `self` is a negative number.
+func Sqrt(self float32) float32 {
+	if self >= 0.0 {
+		return FromBits((ToBits(self) + 0x3f80_0000) >> 1)
+	} else {
+		return NaN
+	}
+}
+
 // Returns the integer part of a number.
 func Trunc(self float32) float32 {
 	const MANTISSA_MASK = 0b0000_0000_0111_1111_1111_1111_1111_1111

arch_tinygo.go

@@ -0,0 +1,33 @@
+//go:build tinygo.wasm
+
+package tinymath
+
+// Functions in this file are inlined and optimized by TinyGo compiler.
+// The result is a single wasm instruction.
+//
+// https://github.com/tinygo-org/tinygo/blob/6384ecace093df2d0b93915886954abfc4ecfe01/compiler/intrinsics.go#L114C5-L114C22
+
+import (
+	"math"
+	"math/bits"
+)
+
+func Ceil(self float32) float32 {
+	return float32(math.Ceil(float64(self)))
+}
+
+func Floor(self float32) float32 {
+	return float32(math.Floor(float64(self)))
+}
+
+func Sqrt(self float32) float32 {
+	return float32(math.Sqrt(float64(self)))
+}
+
+func Trunc(self float32) float32 {
+	return float32(math.Trunc(float64(self)))
+}
+
+func leadingZeros(x uint32) uint32 {
+	return uint32(bits.LeadingZeros32(x))
+}

size_bench/std/atan.go

@@ -0,0 +1,10 @@
+//go:build !none || atan
+
+package main
+
+import "math"
+
+//go:export f
+func Atan(a float64) float64 {
+	return math.Atan(a)
+}

size_bench/std/sqrt.go

@@ -0,0 +1,10 @@
+//go:build !none || sqrt
+
+package main
+
+import "math"
+
+//go:export f
+func Sqrt(x float64) float64 {
+	return math.Sqrt(x)
+}

size_bench/std/tan.go

@@ -0,0 +1,10 @@
+//go:build !none || tan
+
+package main
+
+import "math"
+
+//go:export f
+func Tan(x float64) float64 {
+	return math.Tan(x)
+}

size_bench/tiny/atan.go

@@ -0,0 +1,10 @@
+//go:build !none || atan
+
+package main
+
+import "github.com/orsinium-labs/tinymath"
+
+//go:export f
+func Atan(a float32) float32 {
+	return tinymath.Atan(a)
+}

size_bench/tiny/sqrt.go

@@ -0,0 +1,10 @@
+//go:build !none || sqrt
+
+package main
+
+import "github.com/orsinium-labs/tinymath"
+
+//go:export f
+func Sqrt(x float32) float32 {
+	return tinymath.Sqrt(x)
+}

size_bench/tiny/tan.go

@@ -0,0 +1,10 @@
+//go:build !none || tan
+
+package main
+
+import "github.com/orsinium-labs/tinymath"
+
+//go:export f
+func Tan(x float32) float32 {
+	return tinymath.Tan(x)
+}

tinymath.go

@@ -349,17 +349,6 @@ func Sign(self float32) float32 {
 	}
 }
 
-// Approximates the square root of a number with an average deviation of ~5%.
-//
-// Returns [`NAN`] if `self` is a negative number.
-func Sqrt(self float32) float32 {
-	if self >= 0.0 {
-		return FromBits((ToBits(self) + 0x3f80_0000) >> 1)
-	} else {
-		return NaN
-	}
-}
-
 func extractExponentBits(self float32) uint32 {
 	return (ToBits(self) & expMask) >> mantissaBits
 }