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
}