DeviceAPICalls.h

Go to the documentation of this file.

#ifdef DFTEFE_WITH_DEVICE
 
#  ifndef dftefeDeviceAPICalls_H
#    define dftefeDeviceAPICalls_H
 
#    include "TypeConfig.h"
 
namespace dftefe
{
  namespace utils
  {
    void
    deviceGetDeviceCount(int *count);
 
    void
    deviceGetDevice(int *deviceId);
 
    void
    deviceSetDevice(int deviceId);
 
    void
    deviceMalloc(void **devPtr, size_type size);
 
    void
    deviceMemset(void *devPtr, size_type count);
 
    template <typename ValueType>
    void
    deviceSetValue(ValueType *devPtr, ValueType value, size_type size);
 
    void
    deviceFree(void *devPtr);
 
    void
    hostPinnedMalloc(void **hostPtr, size_type size);
 
    void
    hostPinnedFree(void *hostPtr);
 
    void
    deviceMemcpyD2H(void *dst, const void *src, size_type count);
 
    void
    deviceMemcpyD2D(void *dst, const void *src, size_type count);
 
    void
    deviceMemcpyH2D(void *dst, const void *src, size_type count);
  } // namespace utils
} // namespace dftefe
 
#  endif // dftefeDeviceAPICalls_H
#endif   // DFTEFE_WITH_DEVICE