From 49e07477b22e44f684a228437ded2fee8f5d79e8 Mon Sep 17 00:00:00 2001 From: rowanc1 Date: Mon, 19 May 2014 01:23:34 -0700 Subject: [PATCH] some cpu optimizations --- simpegEM/TDEM/TDEM_b.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/simpegEM/TDEM/TDEM_b.py b/simpegEM/TDEM/TDEM_b.py index a9acc181..4463cef6 100644 --- a/simpegEM/TDEM/TDEM_b.py +++ b/simpegEM/TDEM/TDEM_b.py @@ -1,5 +1,5 @@ from BaseTDEM import BaseTDEMProblem, FieldsTDEM -from SimPEG.Utils import mkvc +from SimPEG.Utils import mkvc, sdiag import numpy as np from SurveyTDEM import SurveyTDEM @@ -110,8 +110,9 @@ class ProblemTDEM_b(BaseTDEMProblem): for i in range(1,self.nT+1): # TODO: G[1] may be dependent on the model # for a galvanic source (deriv of the dc problem) - for tx in self.survey.txList: - p[tx, 'e', i] = -u[tx,'e',i]*c # i.e.: - diag(e) * MsigDeriv * v + # + # Do multiplication for all tx in self.survey.txList + p[:, 'e', i] = - sdiag(c) * u[:,'e',i] # i.e.: - sdiag(MsigDeriv * v) * e return p def Gtvec(self, m, vec, u=None):