GinkgoRBFKernels.hpp

Go to the documentation of this file.

#pragma once
 
#include "mapping/GinkgoDefinitions.hpp"
#include "mapping/impl/BasisFunctions.hpp"
 
namespace precice {
namespace mapping {
 
std::shared_ptr<gko::Executor> create_device_executor(const std::string &execName, bool enableUnifiedMemory);
 
namespace kernel {
 
template <typename EvalFunctionType>
void create_rbf_system_matrix(std::shared_ptr<const gko::Executor> exec,
                              bool                                 unifiedMemory,
                              gko::ptr_param<GinkgoMatrix> mtx, const std::array<bool, 3> activeAxis,
                              gko::ptr_param<GinkgoMatrix> supportPoints, gko::ptr_param<GinkgoMatrix> targetPoints,
                              EvalFunctionType f, ::precice::mapping::RadialBasisParameters rbf_params, bool addPolynomial, unsigned int extraDims = 0);
 
void fill_polynomial_matrix(std::shared_ptr<const gko::Executor> exec,
                            bool                                 unifiedMemory,
                            gko::ptr_param<GinkgoMatrix> mtx, gko::ptr_param<const GinkgoMatrix> x, const unsigned int dims);
 
} // namespace kernel
} // namespace mapping
} // namespace precice