def _compute_totals(self, of=None, wrt=None, return_format='flat_dict', global_names=True): """ Compute derivatives of desired quantities with respect to desired inputs. All derivatives are returned using driver scaling. Parameters ---------- of : list of variable name strings or None Variables whose derivatives will be computed. Default is None, which uses the driver's objectives and constraints. wrt : list of variable name strings or None Variables with respect to which the derivatives will be computed. Default is None, which uses the driver's desvars. return_format : string Format to return the derivatives. Default is a 'flat_dict', which returns them in a dictionary whose keys are tuples of form (of, wrt). For the scipy optimizer, 'array' is also supported. global_names : bool Set to True when passing in global names to skip some translation steps. Returns ------- derivs : object Derivatives in form requested by 'return_format'. """ total_jac = self._total_jac debug_print = 'totals' in self.options['debug_print'] and (not MPI or MPI.COMM_WORLD.rank == 0) if debug_print: header = 'Driver total derivatives for iteration: ' + str(self.iter_count) print(header) print(len(header) * '-' + '\n') if self._problem.model._owns_approx_jac: if total_jac is None: self._total_jac = total_jac = _TotalJacInfo(self._problem, of, wrt, global_names, return_format, approx=True, debug_print=debug_print) return total_jac.compute_totals_approx() else: if total_jac is None: total_jac = _TotalJacInfo(self._problem, of, wrt, global_names, return_format, debug_print=debug_print) # don't cache linear constraint jacobian if not total_jac.has_lin_cons: self._total_jac = total_jac return total_jac.compute_totals()
def _compute_totals(self, of=None, wrt=None, return_format='flat_dict', global_names=None, use_abs_names=True): """ Compute derivatives of desired quantities with respect to desired inputs. All derivatives are returned using driver scaling. Parameters ---------- of : list of variable name strings or None Variables whose derivatives will be computed. Default is None, which uses the driver's objectives and constraints. wrt : list of variable name strings or None Variables with respect to which the derivatives will be computed. Default is None, which uses the driver's desvars. return_format : string Format to return the derivatives. Default is a 'flat_dict', which returns them in a dictionary whose keys are tuples of form (of, wrt). For the scipy optimizer, 'array' is also supported. global_names : bool Deprecated. Use 'use_abs_names' instead. use_abs_names : bool Set to True when passing in absolute names to skip some translation steps. Returns ------- derivs : object Derivatives in form requested by 'return_format'. """ problem = self._problem() total_jac = self._total_jac debug_print = 'totals' in self.options['debug_print'] and ( not MPI or problem.comm.rank == 0) if debug_print: header = 'Driver total derivatives for iteration: ' + str( self.iter_count) print(header) print(len(header) * '-' + '\n') if global_names is not None: warn_deprecation( "'global_names' is deprecated in calls to _compute_totals. " "Use 'use_abs_names' instead.") use_abs_names = global_names if problem.model._owns_approx_jac: self._recording_iter.push(('_compute_totals_approx', 0)) try: if total_jac is None: total_jac = _TotalJacInfo(problem, of, wrt, use_abs_names, return_format, approx=True, debug_print=debug_print) # Don't cache linear constraint jacobian if not total_jac.has_lin_cons: self._total_jac = total_jac totals = total_jac.compute_totals_approx(initialize=True) else: totals = total_jac.compute_totals_approx() finally: self._recording_iter.pop() else: if total_jac is None: total_jac = _TotalJacInfo(problem, of, wrt, use_abs_names, return_format, debug_print=debug_print) # don't cache linear constraint jacobian if not total_jac.has_lin_cons: self._total_jac = total_jac self._recording_iter.push(('_compute_totals', 0)) try: totals = total_jac.compute_totals() finally: self._recording_iter.pop() if self._rec_mgr._recorders and self.recording_options[ 'record_derivatives']: metadata = create_local_meta(self._get_name()) total_jac.record_derivatives(self, metadata) return totals
def _compute_totals(self, of=None, wrt=None, return_format='flat_dict', global_names=True): """ Compute derivatives of desired quantities with respect to desired inputs. All derivatives are returned using driver scaling. Parameters ---------- of : list of variable name strings or None Variables whose derivatives will be computed. Default is None, which uses the driver's objectives and constraints. wrt : list of variable name strings or None Variables with respect to which the derivatives will be computed. Default is None, which uses the driver's desvars. return_format : string Format to return the derivatives. Default is a 'flat_dict', which returns them in a dictionary whose keys are tuples of form (of, wrt). For the scipy optimizer, 'array' is also supported. global_names : bool Set to True when passing in global names to skip some translation steps. Returns ------- derivs : object Derivatives in form requested by 'return_format'. """ problem = self._problem total_jac = self._total_jac debug_print = 'totals' in self.options['debug_print'] and (not MPI or MPI.COMM_WORLD.rank == 0) if debug_print: header = 'Driver total derivatives for iteration: ' + str(self.iter_count) print(header) print(len(header) * '-' + '\n') if problem.model._owns_approx_jac: self._recording_iter.stack.append(('_compute_totals_approx', 0)) try: if total_jac is None: total_jac = _TotalJacInfo(problem, of, wrt, global_names, return_format, approx=True, debug_print=debug_print) self._total_jac = total_jac totals = total_jac.compute_totals_approx(initialize=True) else: totals = total_jac.compute_totals_approx() finally: self._recording_iter.stack.pop() else: if total_jac is None: total_jac = _TotalJacInfo(problem, of, wrt, global_names, return_format, debug_print=debug_print) # don't cache linear constraint jacobian if not total_jac.has_lin_cons: self._total_jac = total_jac self._recording_iter.stack.append(('_compute_totals', 0)) try: totals = total_jac.compute_totals() finally: self._recording_iter.stack.pop() if self._rec_mgr._recorders and self.recording_options['record_derivatives']: metadata = create_local_meta(self._get_name()) total_jac.record_derivatives(self, metadata) return totals