1 // Copyright ©2015 The Gonum Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style
3 // license that can be found in the LICENSE file.
10 "golang.org/x/exp/rand"
12 "gonum.org/v1/gonum/blas"
13 "gonum.org/v1/gonum/blas/blas64"
14 "gonum.org/v1/gonum/floats"
17 type Dorm2rer interface {
19 Dorm2r(side blas.Side, trans blas.Transpose, m, n, k int, a []float64, lda int, tau, c []float64, ldc int, work []float64)
22 func Dorm2rTest(t *testing.T, impl Dorm2rer) {
23 rnd := rand.New(rand.NewSource(1))
24 for _, side := range []blas.Side{blas.Left, blas.Right} {
25 for _, trans := range []blas.Transpose{blas.NoTrans, blas.Trans} {
26 for _, test := range []struct {
27 common, adim, cdim, lda, ldc int
48 var ma, na, mc, nc int
49 if side == blas.Left {
61 // Generate a random matrix
66 a := make([]float64, ma*lda)
74 // Compute random C matrix
75 c := make([]float64, mc*ldc)
82 tau := make([]float64, k)
83 work := make([]float64, 1)
84 impl.Dgeqrf(ma, na, a, lda, tau, work, -1)
85 work = make([]float64, int(work[0]))
86 impl.Dgeqrf(ma, na, a, lda, tau, work, len(work))
88 // Build Q from result
89 q := constructQ("QR", ma, na, a, lda, tau)
91 cMat := blas64.General{
95 Data: make([]float64, len(c)),
98 cMatCopy := blas64.General{
102 Data: make([]float64, len(cMat.Data)),
104 copy(cMatCopy.Data, cMat.Data)
108 case side == blas.Left && trans == blas.NoTrans:
109 blas64.Gemm(blas.NoTrans, blas.NoTrans, 1, q, cMatCopy, 0, cMat)
110 case side == blas.Left && trans == blas.Trans:
111 blas64.Gemm(blas.Trans, blas.NoTrans, 1, q, cMatCopy, 0, cMat)
112 case side == blas.Right && trans == blas.NoTrans:
113 blas64.Gemm(blas.NoTrans, blas.NoTrans, 1, cMatCopy, q, 0, cMat)
114 case side == blas.Right && trans == blas.Trans:
115 blas64.Gemm(blas.NoTrans, blas.Trans, 1, cMatCopy, q, 0, cMat)
117 // Do Dorm2r ard compare
118 if side == blas.Left {
119 work = make([]float64, nc)
121 work = make([]float64, mc)
123 aCopy := make([]float64, len(a))
125 tauCopy := make([]float64, len(tau))
127 impl.Dorm2r(side, trans, mc, nc, k, a, lda, tau, c, ldc, work)
128 if !floats.Equal(a, aCopy) {
129 t.Errorf("a changed in call")
131 if !floats.Equal(tau, tauCopy) {
132 t.Errorf("tau changed in call")
134 if !floats.EqualApprox(cMat.Data, c, 1e-14) {
135 t.Errorf("Multiplication mismatch.\n Want %v \n got %v.", cMat.Data, c)