MooseInit.C

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html

#include "MooseInit.h"
#include "ParallelUniqueId.h"
#include "Factory.h"
#include "ActionFactory.h"
#include "Executioner.h"
#include "MooseRandom.h"

#include "libmesh/petsc_solver_exception.h"
#include "libmesh/reference_counter.h"

// PETSc
#include "petscsys.h"

#ifdef LIBMESH_HAVE_OPENMP
#include <omp.h>
#endif

#ifdef MOOSE_LIBTORCH_ENABLED
#include <ATen/Parallel.h>
#endif

#include <unistd.h>
#include <signal.h>

void
SigHandler(int signum)
{
  Moose::interrupt_signal_number = signum;
  return;
}

void
RegisterSigHandler()
{
  signal(SIGUSR1, SigHandler);
}

MooseInit::MooseInit(int argc, char * argv[], MPI_Comm COMM_WORLD_IN)
  : LibMeshInit(argc, argv, COMM_WORLD_IN)
{
  LibmeshPetscCallA(COMM_WORLD_IN, PetscPopSignalHandler()); // get rid of PETSc error handler

// Set the number of OpenMP threads to the same as the number of threads libMesh is going to use
#ifdef LIBMESH_HAVE_OPENMP
  omp_set_num_threads(libMesh::n_threads());
#endif

#ifdef MOOSE_LIBTORCH_ENABLED
  at::set_num_threads(libMesh::n_threads());
  at::set_num_interop_threads(libMesh::n_threads());
#endif

  if (!libMesh::on_command_line("--enable-refcount-printing"))
    libMesh::ReferenceCounter::disable_print_counter_info();

  ParallelUniqueId::initialize();

  // Make sure that any calls to the global random number generator are consistent among processes
  MooseRandom::seed(0);

  RegisterSigHandler();

#ifdef MOOSE_KOKKOS_ENABLED
  initKokkos();
#endif
}