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

blas: add hermitian conversions

Browse Source
This commit is contained in:
kortschak
2017-08-02 10:07:23 +09:30
committed by Dan Kortschak
parent f4f1896531
commit 151f028ba7
19 changed files with 2403 additions and 1760 deletions
Download Patch File Download Diff File Expand all files Collapse all files

146
blas/blas32/conv.go
View File

@@ -45,65 +45,6 @@ func (t General) From(a GeneralCols) {
}
}
// SymmetricCols represents a matrix using the conventional column-major storage scheme.
type SymmetricCols Symmetric
// From fills the receiver with elements from a. The receiver
// must have the same dimensions and uplo as a and have adequate
// backing data storage.
func (t SymmetricCols) From(a Symmetric) {
if t.N != a.N {
panic("blas32: mismatched dimension")
}
if t.Uplo != a.Uplo {
panic("blas32: mismatched BLAS uplo")
}
switch a.Uplo {
default:
panic("blas32: bad BLAS uplo")
case blas.Upper:
for i := 0; i < a.N; i++ {
for j := i; j < a.N; j++ {
t.Data[i+j*t.Stride] = a.Data[i*a.Stride+j]
}
}
case blas.Lower:
for i := 0; i < a.N; i++ {
for j := 0; j <= i; j++ {
t.Data[i+j*t.Stride] = a.Data[i*a.Stride+j]
}
}
}
}
// From fills the receiver with elements from a. The receiver
// must have the same dimensions and uplo as a and have adequate
// backing data storage.
func (t Symmetric) From(a SymmetricCols) {
if t.N != a.N {
panic("blas32: mismatched dimension")
}
if t.Uplo != a.Uplo {
panic("blas32: mismatched BLAS uplo")
}
switch a.Uplo {
default:
panic("blas32: bad BLAS uplo")
case blas.Upper:
for i := 0; i < a.N; i++ {
for j := i; j < a.N; j++ {
t.Data[i*t.Stride+j] = a.Data[i+j*a.Stride]
}
}
case blas.Lower:
for i := 0; i < a.N; i++ {
for j := 0; j <= i; j++ {
t.Data[i*t.Stride+j] = a.Data[i+j*a.Stride]
}
}
}
}
// TriangularCols represents a matrix using the conventional column-major storage scheme.
type TriangularCols Triangular
@@ -230,93 +171,6 @@ func (t Band) From(a BandCols) {
}
}
// SymmetricBandCols represents a symmetric matrix using the band column-major storage scheme.
type SymmetricBandCols SymmetricBand
// From fills the receiver with elements from a. The receiver
// must have the same dimensions, bandwidth and uplo as a and
// have adequate backing data storage.
func (t SymmetricBandCols) From(a SymmetricBand) {
if t.N != a.N {
panic("blas32: mismatched dimension")
}
if t.K != a.K {
panic("blas32: mismatched bandwidth")
}
if a.Stride < a.K+1 {
panic("blas32: short stride for source")
}
if t.Stride < t.K+1 {
panic("blas32: short stride for destination")
}
if t.Uplo != a.Uplo {
panic("blas32: mismatched BLAS uplo")
}
dst := BandCols{
Rows: t.N, Cols: t.N,
Stride: t.Stride,
Data: t.Data,
}
src := Band{
Rows: a.N, Cols: a.N,
Stride: a.Stride,
Data: a.Data,
}
switch a.Uplo {
default:
panic("blas32: bad BLAS uplo")
case blas.Upper:
dst.KU = t.K
src.KU = a.K
case blas.Lower:
dst.KL = t.K
src.KL = a.K
}
dst.From(src)
}
// From fills the receiver with elements from a. The receiver
// must have the same dimensions, bandwidth and uplo as a and
// have adequate backing data storage.
func (t SymmetricBand) From(a SymmetricBandCols) {
if t.N != a.N {
panic("blas32: mismatched dimension")
}
if t.K != a.K {
panic("blas32: mismatched bandwidth")
}
if a.Stride < a.K+1 {
panic("blas32: short stride for source")
}
if t.Stride < t.K+1 {
panic("blas32: short stride for destination")
}
if t.Uplo != a.Uplo {
panic("blas32: mismatched BLAS uplo")
}
dst := Band{
Rows: t.N, Cols: t.N,
Stride: t.Stride,
Data: t.Data,
}
src := BandCols{
Rows: a.N, Cols: a.N,
Stride: a.Stride,
Data: a.Data,
}
switch a.Uplo {
default:
panic("blas32: bad BLAS uplo")
case blas.Upper:
dst.KU = t.K
src.KU = a.K
case blas.Lower:
dst.KL = t.K
src.KL = a.K
}
dst.From(src)
}
// TriangularBandCols represents a symmetric matrix using the band column-major storage scheme.
type TriangularBandCols TriangularBand