Eigen::DefaultDevice Struct Reference

Public Member Functions
void *	allocate (size_t num_bytes) const
void *	allocate_temp (size_t num_bytes) const
void	deallocate (void *buffer) const
void	deallocate_temp (void *buffer) const
template<typename T >
void	fill (T *begin, T *end, const T &value) const
size_t	firstLevelCacheSize () const
template<typename Type >
Type	get (Type data) const
size_t	lastLevelCacheSize () const
int	majorDeviceVersion () const
void	memcpy (void *dst, const void *src, size_t n) const
void	memcpyDeviceToHost (void *dst, const void *src, size_t n) const
void	memcpyHostToDevice (void *dst, const void *src, size_t n) const
void	memset (void *buffer, int c, size_t n) const
size_t	numThreads () const
void	synchronize () const

Detailed Description

Definition at line 19 of file TensorDeviceDefault.h.

Member Function Documentation

allocate()

void* Eigen::DefaultDevice::allocate ( size_t  num_bytes ) const

inline

Definition at line 20 of file TensorDeviceDefault.h.

                                                                               {
    return internal::aligned_malloc(num_bytes);
  }

allocate_temp()

void* Eigen::DefaultDevice::allocate_temp ( size_t  num_bytes ) const

inline

Definition at line 26 of file TensorDeviceDefault.h.

                                                                                      {
    return allocate(num_bytes);
  }

deallocate()

void Eigen::DefaultDevice::deallocate ( void *  buffer ) const

inline

Definition at line 23 of file TensorDeviceDefault.h.

                                                                            {
    internal::aligned_free(buffer);
  }

deallocate_temp()

void Eigen::DefaultDevice::deallocate_temp ( void *  buffer ) const

inline

Definition at line 29 of file TensorDeviceDefault.h.

                                                                                 {
    deallocate(buffer);
  }

fill()

template<typename T >

void Eigen::DefaultDevice::fill ( T *  begin, T *  end, const T &  value  ) const

inline

Definition at line 45 of file TensorDeviceDefault.h.

                                                                                          {
#ifdef EIGEN_GPU_COMPILE_PHASE
    // std::fill is not a device function, so resort to simple loop.
    for (T* it = begin; it != end; ++it) {
      *it = value;
    }
#else
    std::fill(begin, end, value);
#endif
  }

firstLevelCacheSize()

size_t Eigen::DefaultDevice::firstLevelCacheSize ( ) const

inline

Definition at line 73 of file TensorDeviceDefault.h.

                                                                           {
#if !defined(EIGEN_GPU_COMPILE_PHASE) && !defined(SYCL_DEVICE_ONLY)
    // Running on the host CPU
    return l1CacheSize();
#elif defined(EIGEN_HIP_DEVICE_COMPILE)
    // Running on a HIP device
    return 48*1024; // FIXME : update this number for HIP
#else
    // Running on a CUDA device, return the amount of shared memory available.
    return 48*1024;
#endif
  }

get()

template<typename Type >

Type Eigen::DefaultDevice::get ( Type  data ) const

inline

Definition at line 56 of file TensorDeviceDefault.h.

                                                                  { 
    return data;
  }

lastLevelCacheSize()

size_t Eigen::DefaultDevice::lastLevelCacheSize ( ) const

inline

Definition at line 86 of file TensorDeviceDefault.h.

                                                                          {
#if !defined(EIGEN_GPU_COMPILE_PHASE) && !defined(SYCL_DEVICE_ONLY)
    // Running single threaded on the host CPU
    return l3CacheSize();
#elif defined(EIGEN_HIP_DEVICE_COMPILE)
    // Running on a HIP device
    return firstLevelCacheSize(); // FIXME : update this number for HIP
#else
    // Running on a CUDA device
    return firstLevelCacheSize();
#endif
  }

majorDeviceVersion()

int Eigen::DefaultDevice::majorDeviceVersion ( ) const

inline

Definition at line 103 of file TensorDeviceDefault.h.

                                                                       {
#if !defined(EIGEN_GPU_COMPILE_PHASE)
    // Running single threaded on the host CPU
    // Should return an enum that encodes the ISA supported by the CPU
    return 1;
#elif defined(EIGEN_HIP_DEVICE_COMPILE)
    // Running on a HIP device
    // return 1 as major for HIP
    return 1;
#else
    // Running on a CUDA device
    return EIGEN_CUDA_ARCH / 100;
#endif
  }

memcpy()

void Eigen::DefaultDevice::memcpy ( void *  dst, const void *  src, size_t  n  ) const

inline

Definition at line 32 of file TensorDeviceDefault.h.

                                                                                                {
    ::memcpy(dst, src, n);
  }

memcpyDeviceToHost()

void Eigen::DefaultDevice::memcpyDeviceToHost ( void *  dst, const void *  src, size_t  n  ) const

inline

Definition at line 38 of file TensorDeviceDefault.h.

                                                                                                            {
    memcpy(dst, src, n);
  }

memcpyHostToDevice()

void Eigen::DefaultDevice::memcpyHostToDevice ( void *  dst, const void *  src, size_t  n  ) const

inline

Definition at line 35 of file TensorDeviceDefault.h.

                                                                                                            {
    memcpy(dst, src, n);
  }

memset()

void Eigen::DefaultDevice::memset ( void *  buffer, int  c, size_t  n  ) const

inline

Definition at line 41 of file TensorDeviceDefault.h.

                                                                                         {
    ::memset(buffer, c, n);
  }

numThreads()

size_t Eigen::DefaultDevice::numThreads ( ) const

inline

Definition at line 60 of file TensorDeviceDefault.h.

                                                                  {
#if !defined(EIGEN_GPU_COMPILE_PHASE)
    // Running on the host CPU
    return 1;
#elif defined(EIGEN_HIP_DEVICE_COMPILE)
    // Running on a HIP device
    return 64;
#else
    // Running on a CUDA device
    return 32;
#endif
  }

synchronize()

void Eigen::DefaultDevice::synchronize ( ) const

inline

Definition at line 99 of file TensorDeviceDefault.h.

                                                                 {
    // Nothing.  Default device operations are synchronous.
  }

---

The documentation for this struct was generated from the following file:

• TensorDeviceDefault.h