nlpodyssey/spago

View on GitHub
mat/internal/f64/asm64/axpy.go

Summary

Maintainability
A
35 mins
Test Coverage
// Copyright ©2015 The Gonum Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

//go:build !amd64 || noasm || gccgo || safe
// +build !amd64 noasm gccgo safe

package asm64

// AxpyUnitary is
//
//    for i, v := range x {
//        y[i] += alpha * v
//    }
func AxpyUnitary(alpha float64, x, y []float64) {
    for i, v := range x {
        y[i] += alpha * v
    }
}

// AxpyUnitaryTo is
//
//    for i, v := range x {
//        dst[i] = alpha*v + y[i]
//    }
func AxpyUnitaryTo(dst []float64, alpha float64, x, y []float64) {
    for i, v := range x {
        dst[i] = alpha*v + y[i]
    }
}

// AxpyInc is
//
//    for i := 0; i < int(n); i++ {
//        y[iy] += alpha * x[ix]
//        ix += incX
//        iy += incY
//    }
func AxpyInc(alpha float64, x, y []float64, n, incX, incY, ix, iy uintptr) {
    for i := 0; i < int(n); i++ {
        y[iy] += alpha * x[ix]
        ix += incX
        iy += incY
    }
}

// AxpyIncTo is
//
//    for i := 0; i < int(n); i++ {
//        dst[idst] = alpha*x[ix] + y[iy]
//        ix += incX
//        iy += incY
//        idst += incDst
//    }
func AxpyIncTo(dst []float64, incDst, idst uintptr, alpha float64, x, y []float64, n, incX, incY, ix, iy uintptr) {
    for i := 0; i < int(n); i++ {
        dst[idst] = alpha*x[ix] + y[iy]
        ix += incX
        iy += incY
        idst += incDst
    }
}