package/gonum
1
0
Fork 0
mirror of https://github.com/gonum/gonum.git synced 2025-10-06 23:52:47 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

mat: don't clobber optimisation for TriDense matrix input

Browse Source
This commit is contained in:
Dan Kortschak
2020-09-14 09:15:40 +09:30
parent f883758f08
commit c62dc81ca3
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
mat/vector.go
View File

@@ -629,6 +629,7 @@ func (v *VecDense) MulVec(a Matrix, b Vector) {
return return
} }
case *TriDense: case *TriDense:
if fast {
v.CopyVec(b) v.CopyVec(b)
aU.checkOverlap(v.asGeneral()) aU.checkOverlap(v.asGeneral())
ta := blas.NoTrans ta := blas.NoTrans
@@ -636,6 +637,8 @@ func (v *VecDense) MulVec(a Matrix, b Vector) {
ta = blas.Trans ta = blas.Trans
} }
blas64.Trmv(ta, aU.mat, v.mat) blas64.Trmv(ta, aU.mat, v.mat)
return
}
case *Dense: case *Dense:
if fast { if fast {
aU.checkOverlap(v.asGeneral()) aU.checkOverlap(v.asGeneral())