Moose::Kokkos::Utils Namespace Reference

Functions
template<typename T >
KOKKOS_INLINE_FUNCTION const T *	find (const T &target, const T *const begin, const T *const end)
	Find a value in an array. More...
KOKKOS_INLINE_FUNCTION void	choleskySolve (Real *const A, Real *const x, Real *const b, const unsigned int n)
	Perform an in-place linear solve using Cholesky decomposition Matrix and right-hand-side vector are modified after this call. More...

Function Documentation

choleskySolve()

KOKKOS_INLINE_FUNCTION void Moose::Kokkos::Utils::choleskySolve	(	Real *const	A,
		Real *const	x,
		Real *const	b,
		const unsigned int	n
	)

Perform an in-place linear solve using Cholesky decomposition Matrix and right-hand-side vector are modified after this call.

Parameters

A	The row-major matrix
x	The solution vector
b	The right-hand-side vector
n	The system size

Definition at line 62 of file KokkosUtils.h.

Referenced by InitialConditionTempl< T >::compute(), and Moose::Kokkos::AuxKernel::computeElementInternal().

{
  for (unsigned int i = 0; i < n; ++i)
  {
    for (unsigned int j = 0; j <= i; ++j)
    {
      Real sum = A[j + n * i];

      for (unsigned int k = 0; k < j; ++k)
        sum -= A[k + n * i] * A[k + n * j];

      if (i == j)
        A[j + n * i] = ::Kokkos::sqrt(sum);
      else
        A[j + n * i] = sum / A[j + n * j];
    }
  }

  for (unsigned int i = 0; i < n; ++i)
  {
    Real sum = b[i];

    for (unsigned int j = 0; j < i; ++j)
      sum -= A[j + n * i] * b[j];

    b[i] = sum / A[i + n * i];
  }

  for (int i = n - 1; i >= 0; --i)
  {
    Real sum = b[i];

    for (unsigned int j = i + 1; j < n; ++j)
      sum -= A[i + n * j] * b[j];

    x[i] = sum / A[i + n * i];
  }
}

find()

template<typename T >

KOKKOS_INLINE_FUNCTION const T* Moose::Kokkos::Utils::find	(	const T &	target,
		const T *const	begin,
		const T *const	end
	)

Find a value in an array.

Parameters

target	The target value to find
begin	The pointer to the first element of the array
end	The pointer next to the last element of the array

Returns

The pointer to the target element, end if the target element was not found

Definition at line 30 of file KokkosUtils.h.

Referenced by ElementIDOutputAction::act(), AddKernelAction::act(), CombineComponentsMeshes::act(), MeshOnlyAction::act(), ChainControlSetupAction::act(), MaterialOutputAction::act(), ChainControl::addChainControlDataDependency(), Moose::MFEM::CoefficientMap< mfem::MatrixCoefficient, mfem::PWMatrixCoefficient >::addCoefficient(), FormattedTable::addData(), MFEMProblem::addFunction(), DiffusionPhysicsBase::addInitialConditions(), InputParameterWarehouse::addInputParameters(), FEProblemBase::addOutput(), SONDefinitionFormatter::addParameters(), Moose::MFEM::CoefficientMap< mfem::MatrixCoefficient, mfem::PWMatrixCoefficient >::addPiecewiseBlocks(), MaterialPropertyStorage::addProperty(), PhysicsBase::allMeshBlocks(), InputParameters::applyParameters(), InputParameters::applySpecificParameters(), TransientMultiApp::appTransferVector(), PhysicsBase::assignBlocks(), ParsedSubdomainIDsGenerator::assignElemSubdomainID(), ParsedSubdomainMeshGenerator::assignElemSubdomainID(), TimedSubdomainModifier::buildFromFile(), Moose::Builder::buildFullTree(), Moose::Builder::buildJsonSyntaxTree(), BreakMeshByElementGenerator::buildSubdomainRestrictedNodeToElemMap(), MooseMeshUtils::changeBoundaryId(), MeshExtruderGenerator::changeID(), PhysicsBase::checkBlockRestrictionIdentical(), InputParametersChecksUtils< BatchMeshGeneratorAction >::checkBlockwiseConsistency(), ComponentBoundaryConditionInterface::checkBoundaryConditionsAllRequested(), FEProblemBase::checkICRestartError(), InputParametersChecksUtils< BatchMeshGeneratorAction >::checkTwoDVectorParamsNoRespectiveOverlap(), CSG::CSGBase::checkUniverseLinking(), InputParametersChecksUtils< BatchMeshGeneratorAction >::checkVectorParamsNoOverlap(), SampledOutput::cloneMesh(), MooseMeshElementConversionUtils::convertHex8Elem(), MooseMeshElementConversionUtils::convertPrism6Elem(), CyclicDependencyException< T, Compare >::CyclicDependencyException(), FunctionMaterialPropertyDescriptor< is_ad >::dependsOn(), SideSetsGeneratorBase::elementSubdomainIdInList(), MultiAppGeneralFieldTransfer::examineLocalValueConflicts(), MultiAppGeneralFieldTransfer::examineReceivedValueConflicts(), MultiAppNearestNodeTransfer::execute(), FEProblemBase::FEProblemBase(), FEProblemSolve::FEProblemSolve(), ComputeFVFluxThread< RangeType, AttribMatrixTags >::finalizeContainers(), HashMap< unsigned, MooseVariableFV *>::find(), Moose::Kokkos::Matrix::find(), NearestNodeLocator::findNodes(), ElementSubdomainModifierBase::findReinitializedElemsAndNodes(), SideSetsGeneratorBase::flood(), CommandLine::formatEntry(), FunctionMaterialBase< is_ad >::FunctionMaterialBase(), DerivativeParsedMaterialHelperTempl< is_ad >::functionsPostParse(), ElementSubdomainModifierBase::gatherPatchElements(), CircularBoundaryCorrectionGenerator::generate(), CutMeshByLevelSetGeneratorBase::generate(), LayeredBase::getBounds(), MooseUtils::DelimitedFileReaderTempl< T >::getData(), VariableCondensationPreconditioner::getDofColRow(), VariableCondensationPreconditioner::getDofToCondense(), DependencyResolver< std::string >::getSortedValuesSets(), PhysicsBase::getSubdomainIDs(), FEProblemBase::getSystemBase(), CommandLine::hasArgument(), ComponentInitialConditionInterface::hasInitialCondition(), ComponentMaterialPropertyInterface::hasProperty(), SystemBase::hasVariable(), BlockRestrictable::initializeBlockRestrictable(), BoundaryRestrictable::initializeBoundaryRestrictable(), PhysicsBase::initializePhysics(), ReferenceResidualConvergence::initialSetup(), ElementSubdomainModifierBase::initialSetup(), SystemBase::isArrayVariable(), Moose::Kokkos::Mesh::isBoundaryNode(), AttribTagBase::isMatch(), AttribBoundaries::isMatch(), ActivateElementsUserObjectBase::isNewlyActivated(), Moose::PetscSupport::isSNESVI(), SolutionUserObjectBase::isVariableNodal(), MooseMeshXYCuttingUtils::lineRemoverCutElemTri(), MooseMeshXYCuttingUtils::lineRemoverMoveNode(), MultiSystemSolveObject::MultiSystemSolveObject(), FEProblemBase::needBoundaryMaterialOnSide(), FEProblemBase::needInterfaceMaterialOnSide(), FEProblemBase::needInternalNeighborSideMaterial(), NodeSetsGeneratorBase::nodeElementsInExcludedSubdomains(), NodeSetsGeneratorBase::nodeElementsInIncludedSubdomains(), NodeSetsGeneratorBase::nodeInExcludedNodesets(), NodeSetsGeneratorBase::nodeInIncludedNodesets(), JSONOutput::outputReporters(), ParsedVectorRealReductionReporter::ParsedVectorRealReductionReporter(), ParsedVectorVectorRealReductionReporter::ParsedVectorVectorRealReductionReporter(), Parser::parseError(), MooseUtils::pathContains(), geom_utils::pointInPolygon(), TheWarehouse::prepare(), VecStore::query(), TheWarehouse::queryInto(), SolutionUserObjectBase::readExodusII(), MooseApp::recursivelyCreateExecutors(), SubProblem::removeAlgebraicGhostingFunctor(), CommandLine::removeArgument(), SubProblem::removeCouplingGhostingFunctor(), DependencyResolver< std::string >::removeEdge(), DependencyResolver< std::string >::removeEdgesInvolving(), MooseMesh::setCoordSystem(), NodeSetsGeneratorBase::setup(), SideSetsGeneratorBase::setup(), NonlinearSystemBase::setupScalingData(), PhysicsBase::shouldCreateVariable(), BlockDeletionGenerator::shouldDelete(), Moose::sideIntersectedByLine(), MultiIndex< Real >::slice(), ElementSubdomainModifierBase::subdomainIsReinitialized(), MooseMeshElementConversionUtils::tetNodesForHex(), MultiAppDofCopyTransfer::transfer(), MultiAppShapeEvaluationTransfer::transferVariable(), Checkpoint::updateCheckpointFiles(), NearestNodeLocator::updatePatch(), ParsedAux::validateGenericVectorNames(), ParsedMaterialBase::validateVectorNames(), Moose::MFEM::EquationSystem::VectorContainsName(), and Moose::Builder::walk().

{
  if (begin == end)
    return end;

  auto left = begin;
  auto right = end - 1;

  while (left <= right)
  {
    auto mid = left + (right - left) / 2;

    if (*mid == target)
      return mid;
    else if (*mid < target)
      left = mid + 1;
    else
      right = mid - 1;
  }

  return end;
}