mirror of
https://github.com/gonum/gonum.git
synced 2025-10-25 08:10:28 +08:00
Working implementation of blocked QR
Improved function documentation Fixed dlarfb and dlarft and added full tests Added dgelq2 Working Dgels Fix many comments and tests Many PR comment responses Responded to more PR comments Many PR comments
This commit is contained in:
138
testlapack/dorm2r.go
Normal file
138
testlapack/dorm2r.go
Normal file
@@ -0,0 +1,138 @@
|
||||
// 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.
|
||||
|
||||
package testlapack
|
||||
|
||||
import (
|
||||
"math/rand"
|
||||
"testing"
|
||||
|
||||
"github.com/gonum/blas"
|
||||
"github.com/gonum/blas/blas64"
|
||||
"github.com/gonum/floats"
|
||||
)
|
||||
|
||||
type Dorm2rer interface {
|
||||
Dgeqrfer
|
||||
Dorm2r(side blas.Side, trans blas.Transpose, m, n, k int, a []float64, lda int, tau, c []float64, ldc int, work []float64)
|
||||
}
|
||||
|
||||
func Dorm2rTest(t *testing.T, impl Dorm2rer) {
|
||||
for _, side := range []blas.Side{blas.Left, blas.Right} {
|
||||
for _, trans := range []blas.Transpose{blas.NoTrans, blas.Trans} {
|
||||
for _, test := range []struct {
|
||||
common, adim, cdim, lda, ldc int
|
||||
}{
|
||||
{3, 4, 5, 0, 0},
|
||||
{3, 5, 4, 0, 0},
|
||||
{4, 3, 5, 0, 0},
|
||||
{4, 5, 3, 0, 0},
|
||||
{5, 3, 4, 0, 0},
|
||||
{5, 4, 3, 0, 0},
|
||||
{3, 4, 5, 6, 20},
|
||||
{3, 5, 4, 6, 20},
|
||||
{4, 3, 5, 6, 20},
|
||||
{4, 5, 3, 6, 20},
|
||||
{5, 3, 4, 6, 20},
|
||||
{5, 4, 3, 6, 20},
|
||||
{3, 4, 5, 20, 6},
|
||||
{3, 5, 4, 20, 6},
|
||||
{4, 3, 5, 20, 6},
|
||||
{4, 5, 3, 20, 6},
|
||||
{5, 3, 4, 20, 6},
|
||||
{5, 4, 3, 20, 6},
|
||||
} {
|
||||
var ma, na, mc, nc int
|
||||
if side == blas.Left {
|
||||
ma = test.common
|
||||
na = test.adim
|
||||
mc = test.common
|
||||
nc = test.cdim
|
||||
} else {
|
||||
ma = test.common
|
||||
na = test.adim
|
||||
mc = test.cdim
|
||||
nc = test.common
|
||||
}
|
||||
|
||||
// Generate a random matrix
|
||||
lda := test.lda
|
||||
if lda == 0 {
|
||||
lda = na
|
||||
}
|
||||
a := make([]float64, ma*lda)
|
||||
for i := range a {
|
||||
a[i] = rand.Float64()
|
||||
}
|
||||
ldc := test.ldc
|
||||
if ldc == 0 {
|
||||
ldc = nc
|
||||
}
|
||||
// Compute random C matrix
|
||||
c := make([]float64, mc*ldc)
|
||||
for i := range c {
|
||||
c[i] = rand.Float64()
|
||||
}
|
||||
|
||||
// Compute QR
|
||||
k := min(ma, na)
|
||||
tau := make([]float64, k)
|
||||
work := make([]float64, 1)
|
||||
impl.Dgeqrf(ma, na, a, lda, tau, work, -1)
|
||||
work = make([]float64, int(work[0]))
|
||||
impl.Dgeqrf(ma, na, a, lda, tau, work, len(work))
|
||||
|
||||
// Build Q from result
|
||||
q := constructQ("QR", ma, na, a, lda, tau)
|
||||
|
||||
cMat := blas64.General{
|
||||
Rows: mc,
|
||||
Cols: nc,
|
||||
Stride: ldc,
|
||||
Data: make([]float64, len(c)),
|
||||
}
|
||||
copy(cMat.Data, c)
|
||||
cMatCopy := blas64.General{
|
||||
Rows: cMat.Rows,
|
||||
Cols: cMat.Cols,
|
||||
Stride: cMat.Stride,
|
||||
Data: make([]float64, len(cMat.Data)),
|
||||
}
|
||||
copy(cMatCopy.Data, cMat.Data)
|
||||
switch {
|
||||
default:
|
||||
panic("bad test")
|
||||
case side == blas.Left && trans == blas.NoTrans:
|
||||
blas64.Gemm(blas.NoTrans, blas.NoTrans, 1, q, cMatCopy, 0, cMat)
|
||||
case side == blas.Left && trans == blas.Trans:
|
||||
blas64.Gemm(blas.Trans, blas.NoTrans, 1, q, cMatCopy, 0, cMat)
|
||||
case side == blas.Right && trans == blas.NoTrans:
|
||||
blas64.Gemm(blas.NoTrans, blas.NoTrans, 1, cMatCopy, q, 0, cMat)
|
||||
case side == blas.Right && trans == blas.Trans:
|
||||
blas64.Gemm(blas.NoTrans, blas.Trans, 1, cMatCopy, q, 0, cMat)
|
||||
}
|
||||
// Do Dorm2r ard compare
|
||||
if side == blas.Left {
|
||||
work = make([]float64, nc)
|
||||
} else {
|
||||
work = make([]float64, mc)
|
||||
}
|
||||
aCopy := make([]float64, len(a))
|
||||
copy(aCopy, a)
|
||||
tauCopy := make([]float64, len(tau))
|
||||
copy(tauCopy, tau)
|
||||
impl.Dorm2r(side, trans, mc, nc, k, a, lda, tau, c, ldc, work)
|
||||
if !floats.Equal(a, aCopy) {
|
||||
t.Errorf("a changed in call")
|
||||
}
|
||||
if !floats.Equal(tau, tauCopy) {
|
||||
t.Errorf("tau changed in call")
|
||||
}
|
||||
if !floats.EqualApprox(cMat.Data, c, 1e-14) {
|
||||
t.Errorf("Multiplication mismatch.\n Want %v \n got %v.", cMat.Data, c)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user