deviceKernelsGeneric.h

Go to the documentation of this file.

// ---------------------------------------------------------------------
//
// Copyright (c) 2017-2025  The Regents of the University of Michigan and DFT-FE
// authors.
//
// This file is part of the DFT-FE code.
//
// The DFT-FE code is free software; you can use it, redistribute
// it, and/or modify it under the terms of the GNU Lesser General
// Public License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
// The full text of the license can be found in the file LICENSE at
// the top level of the DFT-FE distribution.
//
// ---------------------------------------------------------------------
 
#if defined(DFTFE_WITH_DEVICE)
#  ifndef deviceKernelsGeneric_H_
#    define deviceKernelsGeneric_H_
 
#    include <dftfeDataTypes.h>
#    include <MemorySpaceType.h>
#    include <TypeConfig.h>
#    include <DeviceTypeConfig.h>
 
namespace dftfe
{
  namespace utils
  {
    namespace deviceKernelsGeneric
    {
      void
      setupDevice(const int &mpi_rank);
 
      template <typename ValueTypeComplex, typename ValueTypeReal>
      void
      copyComplexArrToRealArrsDevice(const dftfe::uInt       size,
                                     const ValueTypeComplex *complexArr,
                                     ValueTypeReal          *realArr,
                                     ValueTypeReal          *imagArr);
 
 
      template <typename ValueTypeComplex, typename ValueTypeReal>
      void
      copyRealArrsToComplexArrDevice(const dftfe::uInt    size,
                                     const ValueTypeReal *realArr,
                                     const ValueTypeReal *imagArr,
                                     ValueTypeComplex    *complexArr);
 
 
 
      template <typename ValueType>
      void
      sadd(ValueType        *y,
           ValueType        *x,
           const ValueType   beta,
           const dftfe::uInt size);
 
 
      // This kernel interpolates the nodal data to quad data
      // The function takes the cell level nodal data
      // and interpolates it to the quad data in each cell
      // by multiplying with the shape function
      template <typename ValueType1, typename ValueType2>
      void
      interpolateNodalDataToQuadDevice(
        const dftfe::uInt numDofsPerElem,
        const dftfe::uInt numQuadPoints,
        const dftfe::uInt numVecs,
        const ValueType2 *parentShapeFunc,
        const ValueType1 *mapPointToCellIndex,
        const ValueType1 *mapPointToProcLocal,
        const ValueType1 *mapPointToShapeFuncIndex,
        const ValueType2 *parentNodalValues,
        ValueType2       *quadValues);
 
    } // namespace deviceKernelsGeneric
  }   // namespace utils
} // namespace dftfe
 
#  endif
#endif