Add: Expose L2 distance in Swift (#255)

Closes #203

Author: mikhailchetin

swift/SimSIMD.swift

@@ -4,35 +4,42 @@ public protocol SimSIMD {
     static var dataType: simsimd_datatype_t { get }
     static var cosine: simsimd_metric_dense_punned_t { get }
     static var dotProduct: simsimd_metric_dense_punned_t { get }
+    static var euclidean: simsimd_metric_dense_punned_t { get }
     static var squaredEuclidean: simsimd_metric_dense_punned_t { get }
 }
 
 extension Int8: SimSIMD {
     public static let dataType = simsimd_datatype_i8_k
     public static let cosine = find(kind: simsimd_metric_cosine_k, dataType: dataType)
     public static let dotProduct = find(kind: simsimd_metric_dot_k, dataType: dataType)
+    public static let euclidean = find(kind: simsimd_metric_euclidean_k, dataType: dataType)
     public static let squaredEuclidean = find(kind: simsimd_metric_sqeuclidean_k, dataType: dataType)
 }
 
+#if !arch(x86_64)
 @available(macOS 11.0, iOS 14.0, tvOS 14.0, watchOS 7.0, *)
 extension Float16: SimSIMD {
     public static let dataType = simsimd_datatype_f16_k
     public static let cosine = find(kind: simsimd_metric_cosine_k, dataType: dataType)
     public static let dotProduct = find(kind: simsimd_metric_dot_k, dataType: dataType)
+    public static let euclidean = find(kind: simsimd_metric_euclidean_k, dataType: dataType)
     public static let squaredEuclidean = find(kind: simsimd_metric_sqeuclidean_k, dataType: dataType)
 }
+#endif
 
 extension Float32: SimSIMD {
     public static let dataType = simsimd_datatype_f32_k
     public static let cosine = find(kind: simsimd_metric_cosine_k, dataType: dataType)
     public static let dotProduct = find(kind: simsimd_metric_inner_k, dataType: dataType)
+    public static let euclidean = find(kind: simsimd_metric_euclidean_k, dataType: dataType)
     public static let squaredEuclidean = find(kind: simsimd_metric_sqeuclidean_k, dataType: dataType)
 }
 
 extension Float64: SimSIMD {
     public static let dataType = simsimd_datatype_f64_k
     public static let cosine = find(kind: simsimd_metric_cosine_k, dataType: dataType)
     public static let dotProduct = find(kind: simsimd_metric_dot_k, dataType: dataType)
+    public static let euclidean = find(kind: simsimd_metric_euclidean_k, dataType: dataType)
     public static let squaredEuclidean = find(kind: simsimd_metric_sqeuclidean_k, dataType: dataType)
 }
 
@@ -47,6 +54,11 @@ extension SimSIMD {
         perform(dotProduct, a: a, b: b)
     }
 
+    @inlinable @inline(__always)
+    public static func euclidean<A, B>(_ a: A, _ b: B) -> Double? where A: Sequence, B: Sequence, A.Element == Self, B.Element == Self {
+        perform(euclidean, a: a, b: b)
+    }
+
     @inlinable @inline(__always)
     public static func sqeuclidean<A, B>(_ a: A, _ b: B) -> Double? where A: Sequence, B: Sequence, A.Element == Self, B.Element == Self {
         perform(squaredEuclidean, a: a, b: b)
@@ -64,6 +76,11 @@ extension RandomAccessCollection where Element: SimSIMD {
         Element.dot(self, b)
     }
 
+    @inlinable @inline(__always)
+    public func euclidean<B>(_ b: B) -> Double? where B: Sequence, B.Element == Element {
+        Element.euclidean(self, b)
+    }
+
     @inlinable @inline(__always)
     public func sqeuclidean<B>(_ b: B) -> Double? where B: Sequence, B.Element == Element {
         Element.sqeuclidean(self, b)

swift/Test.swift

@@ -13,12 +13,14 @@ class SimSIMDTests: XCTestCase {
         XCTAssertEqual(result, 0.00012027938, accuracy: 0.01)
     }
 
+    #if !arch(x86_64)
     func testCosineFloat16() throws {
         let a: [Float16] = [1.0, 2.0, 3.0]
         let b: [Float16] = [1.0, 2.0, 3.0]
         let result = try XCTUnwrap(a.cosine(b))
         XCTAssertEqual(result, 0.004930496, accuracy: 0.01)
     }
+    #endif
 
     func testCosineFloat32() throws {
         let a: [Float32] = [1.0, 2.0, 3.0]
@@ -41,12 +43,14 @@ class SimSIMDTests: XCTestCase {
         XCTAssertEqual(result, 32.0, accuracy: 0.01)
     }
 
+    #if !arch(x86_64)
     func testDotFloat16() throws {
         let a: [Float16] = [1.0, 2.0, 3.0]
         let b: [Float16] = [4.0, 5.0, 6.0]
         let result = try XCTUnwrap(a.dot(b))
         XCTAssertEqual(result, 32.0, accuracy: 0.01)
     }
+    #endif
 
     func testDotFloat32() throws {
         let a: [Float32] = [1.0, 2.0, 3.0]
@@ -62,19 +66,51 @@ class SimSIMDTests: XCTestCase {
         XCTAssertEqual(result, 32.0, accuracy: 0.01)
     }
 
+    func testEuclideanInt8() throws {
+        let a: [Int8] = [1, 2, 3]
+        let b: [Int8] = [4, 5, 6]
+        let result = try XCTUnwrap(a.euclidean(b))
+        XCTAssertEqual(result, 5.196152422706632, accuracy: 0.01)
+    }
+
+    #if !arch(x86_64)
+    func testEuclideanFloat16() throws {
+        let a: [Float16] = [1.0, 2.0, 3.0]
+        let b: [Float16] = [4.0, 5.0, 6.0]
+        let result = try XCTUnwrap(a.euclidean(b))
+        XCTAssertEqual(result, 5.196152422706632, accuracy: 0.01)
+    }
+    #endif
+
+    func testEuclideanFloat32() throws {
+        let a: [Float32] = [1.0, 2.0, 3.0]
+        let b: [Float32] = [4.0, 5.0, 6.0]
+        let result = try XCTUnwrap(a.euclidean(b))
+        XCTAssertEqual(result, 5.196152422706632, accuracy: 0.01)
+    }
+
+    func testEuclideanFloat64() throws {
+        let a: [Float64] = [1.0, 2.0, 3.0]
+        let b: [Float64] = [4.0, 5.0, 6.0]
+        let result = try XCTUnwrap(a.euclidean(b))
+        XCTAssertEqual(result, 5.196152422706632, accuracy: 0.01)
+    }
+
     func testSqeuclideanInt8() throws {
         let a: [Int8] = [1, 2, 3]
         let b: [Int8] = [4, 5, 6]
         let result = try XCTUnwrap(a.sqeuclidean(b))
         XCTAssertEqual(result, 27.0, accuracy: 0.01)
     }
 
+    #if !arch(x86_64)
     func testSqeuclideanFloat16() throws {
         let a: [Float16] = [1.0, 2.0, 3.0]
         let b: [Float16] = [4.0, 5.0, 6.0]
         let result = try XCTUnwrap(a.sqeuclidean(b))
         XCTAssertEqual(result, 27.0, accuracy: 0.01)
     }
+    #endif
 
     func testSqeuclideanFloat32() throws {
         let a: [Float32] = [1.0, 2.0, 3.0]