ce5fd63d0a
Co-authored-by: camierjs <camierjs@gmail.com> Co-authored-by: Veselin Dobrev <dobrev@llnl.gov>
67 lines
2.2 KiB
C++
67 lines
2.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.
|
|
|
|
#include "../config/config.hpp"
|
|
|
|
#ifndef MFEM_ENZYME_HPP
|
|
#define MFEM_ENZYME_HPP
|
|
|
|
#ifdef MFEM_USE_ENZYME
|
|
/*
|
|
* Variables prefixed with enzyme_* or function types prefixed with __enzyme_*,
|
|
* are variables which will get preprocessed in the LLVM intermediate
|
|
* representation when the Enzyme LLVM plugin is loaded. See the Enzyme
|
|
* documentation (https://enzyme.mit.edu) for more information.
|
|
*/
|
|
|
|
extern int enzyme_dup;
|
|
extern int enzyme_dupnoneed;
|
|
extern int enzyme_out;
|
|
extern int enzyme_const;
|
|
extern int enzyme_interleave;
|
|
|
|
#if defined(MFEM_USE_CUDA) || defined(MFEM_USE_HIP)
|
|
#define MFEM_DEVICE_EXTERN_STMT(name) extern __device__ int name;
|
|
#else
|
|
#define MFEM_DEVICE_EXTERN_STMT(name)
|
|
#endif
|
|
|
|
MFEM_DEVICE_EXTERN_STMT(enzyme_dup)
|
|
MFEM_DEVICE_EXTERN_STMT(enzyme_dupnoneed)
|
|
MFEM_DEVICE_EXTERN_STMT(enzyme_out)
|
|
MFEM_DEVICE_EXTERN_STMT(enzyme_const)
|
|
MFEM_DEVICE_EXTERN_STMT(enzyme_interleave)
|
|
|
|
// warning: if inlined, triggers function '__enzyme_autodiff' is not defined
|
|
template <typename return_type, typename... Args>
|
|
MFEM_HOST_DEVICE
|
|
return_type __enzyme_autodiff(Args...);
|
|
|
|
// warning: if inlined, triggers function '__enzyme_fwddiff' is not defined
|
|
template <typename return_type, typename... Args>
|
|
MFEM_HOST_DEVICE
|
|
return_type __enzyme_fwddiff(Args...);
|
|
|
|
#define MFEM_ENZYME_INACTIVENOFREE __attribute__((enzyme_inactive, enzyme_nofree))
|
|
#define MFEM_ENZYME_INACTIVE __attribute__((enzyme_inactive))
|
|
#define MFEM_ENZYME_FN_LIKE(x) __attribute__((enzyme_function_like(#x)))
|
|
|
|
#else
|
|
#define MFEM_ENZYME_INACTIVENOFREE
|
|
#define MFEM_ENZYME_INACTIVE
|
|
#define MFEM_ENZYME_FN_LIKE(x)
|
|
#endif
|
|
|
|
#define MFEM_ENZYME_FN_LIKE_FREE MFEM_ENZYME_FN_LIKE(free)
|
|
#define MFEM_ENZYME_FN_LIKE_DYNCAST MFEM_ENZYME_FN_LIKE(__dynamic_cast)
|
|
|
|
#endif
|