Update vdsp implementation for XP

Author: pedrovgs

Sources/Matft/library/vDSP.swift

@@ -2438,34 +2438,57 @@ internal func vDSP_vfixru32D(_ src: UnsafePointer<Double>, _ srcStride: Int, _ d
 
 // MARK: - vDSP Wrapper Functions for WASI
 
+@inline(__always)
+fileprivate func _adjust_ptr<T>(_ ptr: UnsafePointer<T>, _ stride: Int, _ size: Int) -> UnsafePointer<T>{
+    guard stride < 0, size > 0 else { return ptr }
+    return ptr + (size - 1) * stride
+}
+
+@inline(__always)
+fileprivate func _adjust_mptr<T>(_ ptr: UnsafeMutablePointer<T>, _ stride: Int, _ size: Int) -> UnsafeMutablePointer<T>{
+    guard stride < 0, size > 0 else { return ptr }
+    return ptr + (size - 1) * stride
+}
+
 @inline(__always)
 internal func wrap_vDSP_convert<T, U>(_ size: Int, _ srcptr: UnsafePointer<T>, _ srcStride: Int, _ dstptr: UnsafeMutablePointer<U>, _ dstStride: Int, _ vDSP_func: (UnsafePointer<T>, Int, UnsafeMutablePointer<U>, Int, Int) -> Void){
+    let srcptr = _adjust_ptr(srcptr, srcStride, size)
+    let dstptr = _adjust_mptr(dstptr, dstStride, size)
     vDSP_func(srcptr, srcStride, dstptr, dstStride, size)
 }
 
 @inline(__always)
 internal func wrap_vDSP_biopvv<T>(_ size: Int, _ lsrcptr: UnsafePointer<T>, _ lsrcStride: Int, _ rsrcptr: UnsafePointer<T>, _ rsrcStride: Int, _ dstptr: UnsafeMutablePointer<T>, _ dstStride: Int, _ vDSP_func: vDSP_biopvv_func<T>){
+    let lsrcptr = _adjust_ptr(lsrcptr, lsrcStride, size)
+    let rsrcptr = _adjust_ptr(rsrcptr, rsrcStride, size)
+    let dstptr = _adjust_mptr(dstptr, dstStride, size)
     vDSP_func(rsrcptr, rsrcStride, lsrcptr, lsrcStride, dstptr, dstStride, size)
 }
 
 @inline(__always)
 internal func wrap_vDSP_biopvs<T>(_ size: Int, _ srcptr: UnsafePointer<T>, _ srcStride: Int, _ scalar: UnsafePointer<T>, _ dstptr: UnsafeMutablePointer<T>, _ dstStride: Int, _ vDSP_func: vDSP_biopvs_func<T>){
+    let srcptr = _adjust_ptr(srcptr, srcStride, size)
+    let dstptr = _adjust_mptr(dstptr, dstStride, size)
     vDSP_func(srcptr, srcStride, scalar, dstptr, dstStride, size)
 }
 
 @inline(__always)
 internal func wrap_vDSP_biopsv<T>(_ size: Int, _ scalar: UnsafePointer<T>, _ srcptr: UnsafePointer<T>, _ srcStride: Int, _ dstptr: UnsafeMutablePointer<T>, _ dstStride: Int, _ vDSP_func: vDSP_biopsv_func<T>){
+    let srcptr = _adjust_ptr(srcptr, srcStride, size)
+    let dstptr = _adjust_mptr(dstptr, dstStride, size)
     vDSP_func(scalar, srcptr, srcStride, dstptr, dstStride, size)
 }
 
 @inline(__always)
 internal func wrap_vDSP_stats<T>(_ size: Int, _ srcptr: UnsafePointer<T>, _ stride: Int, _ dstptr: UnsafeMutablePointer<T>, _ vDSP_func: vDSP_stats_func<T>){
+    let srcptr = _adjust_ptr(srcptr, stride, size)
     vDSP_func(srcptr, stride, dstptr, size)
 }
 
 @inline(__always)
 internal func wrap_vDSP_stats_index<T: MfStorable>(_ size: Int, _ srcptr: UnsafePointer<T>, _ stride: Int, _ dstptr: UnsafeMutablePointer<UInt>, _ vDSP_func: vDSP_stats_index_func<T>){
     var tmp = Array(repeating: T.zero, count: size)
+    let srcptr = _adjust_ptr(srcptr, stride, size)
     vDSP_func(srcptr, stride, &tmp, dstptr, size)
 }
 
@@ -2489,16 +2512,23 @@ internal func wrap_vDSP_clip<T: MfStorable>(_ size: Int, _ srcptr: UnsafePointer
 
 @inline(__always)
 internal func wrap_vDSP_cmprs<T: MfStorable>(_ size: Int, _ srcptr: UnsafePointer<T>, _ srcStride: Int, _ indptr: UnsafePointer<T>, _ indStride: Int, _ dstptr: UnsafeMutablePointer<T>, _ dstStride: Int, _ vDSP_func: vDSP_vcmprs_func<T>){
+    let srcptr = _adjust_ptr(srcptr, srcStride, size)
+    let indptr = _adjust_ptr(indptr, indStride, size)
+    let dstptr = _adjust_mptr(dstptr, dstStride, size)
     vDSP_func(srcptr, srcStride, indptr, indStride, dstptr, dstStride, size)
 }
 
 @inline(__always)
 internal func wrap_vDSP_gathr<T: MfStorable>(_ size: Int, _ srcptr: UnsafePointer<T>, _ indptr: UnsafePointer<UInt>, _ indStride: Int, _ dstptr: UnsafeMutablePointer<T>, _ dstStride: Int, _ vDSP_func: vDSP_vgathr_func<T>){
+    let indptr = _adjust_ptr(indptr, indStride, size)
+    let dstptr = _adjust_mptr(dstptr, dstStride, size)
     vDSP_func(srcptr, indptr, indStride, dstptr, dstStride, size)
 }
 
 @inline(__always)
 internal func wrap_vDSP_dotpr<T>(_ size: Int, _ lsrcptr: UnsafePointer<T>, _ lsrcStride: Int, _ rsrcptr: UnsafePointer<T>, _ rsrcStride: Int, _ dstptr: UnsafeMutablePointer<T>, _ vDSP_func: vDSP_dotpr_func<T>){
+    let lsrcptr = _adjust_ptr(lsrcptr, lsrcStride, size)
+    let rsrcptr = _adjust_ptr(rsrcptr, rsrcStride, size)
     vDSP_func(lsrcptr, lsrcStride, rsrcptr, rsrcStride, dstptr, size)
 }