Files
mfem/tests/unit/pgpu_unit_test_main.cpp
T
Andrew Ho 8fbfdb19fe added general GPU and CUDA/HIP-specific unit tests
added gpu, raja-gpu, etc. for generic GPU device configuration
2025-03-24 13:23:43 -07:00

38 lines
1.2 KiB
C++

// Copyright (c) 2010-2025, Lawrence Livermore National Security, LLC. Produced
// at the Lawrence Livermore National Laboratory. All Rights reserved. See files
// LICENSE and NOTICE for details. LLNL-CODE-806117.
//
// This file is part of the MFEM library. For more information and source code
// availability visit https://mfem.org.
//
// MFEM is free software; you can redistribute it and/or modify it under the
// terms of the BSD-3 license. We welcome feedback and contributions, see file
// CONTRIBUTING.md for details.
#define CATCH_CONFIG_NOSTDOUT
#define CATCH_CONFIG_RUNNER
#include "mfem.hpp"
#include "run_unit_tests.hpp"
#ifndef MFEM_USE_MPI
#error "This test should be disabled without MFEM_USE_MPI!"
#endif
int main(int argc, char *argv[])
{
#ifdef MFEM_USE_SINGLE
std::cout << "\nThe parallel GPU unit tests are not supported in single"
" precision.\n\n";
return MFEM_SKIP_RETURN_VALUE;
#endif
#ifdef MFEM_USE_MPI
mfem::Mpi::Init();
mfem::Hypre::Init();
#endif
mfem::Device device("gpu");
// Include only tests that are labeled with both CUDA and Parallel.
return RunCatchSession(argc, argv, {"[GPU]","[Parallel]"}, Root());
}