Fix fast path Pow for NaN-filled matrices

gonum · Jan 1, 2015 · 35f9fa1 · 35f9fa1
1 parent 335a841
commit 35f9fa1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/mat64/dense_arithmetic.go b/mat64/dense_arithmetic.go
@@ -528,6 +528,9 @@ func (m *Dense) Pow(a Matrix, n int) {
 		m.Copy(a)
 		return
 	case 2:
+		for i := 0; i < r; i++ {
+			zero(m.mat.Data[i*m.mat.Stride : i*m.mat.Stride+c])
+		}
 		m.Mul(a, a)
 		return
 	}