apollo::perception::base::SyncedMemory类 参考

Manages memory allocation and synchronization between the host (CPU) and device (GPU). 更多...

#include <syncedmem.h>

apollo::perception::base::SyncedMemory 的协作图:

Public 类型
enum	SyncedHead { UNINITIALIZED , HEAD_AT_CPU , HEAD_AT_GPU , SYNCED }

Public 成员函数
	SyncedMemory (bool use_cuda)
	SyncedMemory (size_t size, bool use_cuda)
	SyncedMemory (const SyncedMemory &)=delete
void	operator= (const SyncedMemory &)=delete
	~SyncedMemory ()
const void *	cpu_data ()
void	set_cpu_data (void *data)
const void *	gpu_data ()
void	set_gpu_data (void *data)
void *	mutable_cpu_data ()
void *	mutable_gpu_data ()
SyncedHead	head () const
void	set_head (SyncedHead head)
void	set_head_gpu ()
void	set_head_cpu ()
size_t	size ()

详细描述

Manages memory allocation and synchronization between the host (CPU) and device (GPU).

TODO(dox): more thorough description.

在文件 syncedmem.h 第 99 行定义.

成员枚举类型说明

SyncedHead

enum apollo::perception::base::SyncedMemory::SyncedHead

枚举值
UNINITIALIZED
HEAD_AT_CPU
HEAD_AT_GPU
SYNCED

在文件 syncedmem.h 第 101 行定义.

{ UNINITIALIZED, HEAD_AT_CPU, HEAD_AT_GPU, SYNCED };

构造及析构函数说明

SyncedMemory() [1/3]

apollo::perception::base::SyncedMemory::SyncedMemory ( bool  use_cuda )

explicit

在文件 syncedmem.cc 第 69 行定义.

    : cpu_ptr_(NULL),
      gpu_ptr_(NULL),
      size_(0),
      head_(UNINITIALIZED),
      own_cpu_data_(false),
      cpu_malloc_use_cuda_(use_cuda),
      own_gpu_data_(false),
      device_(-1) {
#if USE_GPU == 1
#ifdef PERCEPTION_DEBUG
  BASE_GPU_CHECK(cudaGetDevice(&device_));
#endif
#endif
}

SyncedMemory() [2/3]

apollo::perception::base::SyncedMemory::SyncedMemory	(	size_t	size,
		bool	use_cuda
	)

在文件 syncedmem.cc 第 85 行定义.

    : cpu_ptr_(NULL),
      gpu_ptr_(NULL),
      size_(size),
      head_(UNINITIALIZED),
      own_cpu_data_(false),
      cpu_malloc_use_cuda_(use_cuda),
      own_gpu_data_(false),
      device_(-1) {
#if USE_GPU == 1
#ifdef PERCEPTION_DEBUG
  BASE_GPU_CHECK(cudaGetDevice(&device_));
#endif
#endif
}

SyncedMemory() [3/3]

apollo::perception::base::SyncedMemory::SyncedMemory ( const SyncedMemory &  )

delete

~SyncedMemory()

apollo::perception::base::SyncedMemory::~SyncedMemory

(

)

在文件 syncedmem.cc 第 101 行定义.

                            {
  check_device();
  if (cpu_ptr_ && own_cpu_data_) {
    PerceptionFreeHost(cpu_ptr_, cpu_malloc_use_cuda_);
  }
 
#if USE_GPU == 1
  if (gpu_ptr_ && own_gpu_data_) {
    BASE_GPU_CHECK(cudaFree(gpu_ptr_));
  }
#endif  // USE_GPU
}

成员函数说明

cpu_data()

const void * apollo::perception::base::SyncedMemory::cpu_data

(

)

在文件 syncedmem.cc 第 172 行定义.

                                   {
  check_device();
  to_cpu();
  return (const void*)cpu_ptr_;
}

gpu_data()

const void * apollo::perception::base::SyncedMemory::gpu_data

(

)

在文件 syncedmem.cc 第 189 行定义.

                                   {
  check_device();
#if USE_GPU == 1
  to_gpu();
  return (const void*)gpu_ptr_;
#else
  NO_GPU;
  return nullptr;
#endif
}

head()

SyncedHead apollo::perception::base::SyncedMemory::head ( ) const

inline

在文件 syncedmem.h 第 116 行定义.

{ return head_; }

mutable_cpu_data()

void * apollo::perception::base::SyncedMemory::mutable_cpu_data

(

)

在文件 syncedmem.cc 第 215 行定义.

                                     {
  check_device();
  to_cpu();
  head_ = HEAD_AT_CPU;
  return cpu_ptr_;
}

mutable_gpu_data()

void * apollo::perception::base::SyncedMemory::mutable_gpu_data

(

)

在文件 syncedmem.cc 第 222 行定义.

                                     {
  check_device();
#if USE_GPU == 1
  to_gpu();
  head_ = HEAD_AT_GPU;
  return gpu_ptr_;
#else
  NO_GPU;
  return nullptr;
#endif
}

operator=()

void apollo::perception::base::SyncedMemory::operator= ( const SyncedMemory &  )

delete

set_cpu_data()

void apollo::perception::base::SyncedMemory::set_cpu_data

(

void *

data

)

在文件 syncedmem.cc 第 178 行定义.

                                          {
  check_device();
  ACHECK(data);
  if (own_cpu_data_) {
    PerceptionFreeHost(cpu_ptr_, cpu_malloc_use_cuda_);
  }
  cpu_ptr_ = data;
  head_ = HEAD_AT_CPU;
  own_cpu_data_ = false;
}

set_gpu_data()

void apollo::perception::base::SyncedMemory::set_gpu_data

(

void *

data

)

在文件 syncedmem.cc 第 200 行定义.

                                          {
  check_device();
#if USE_GPU == 1
  ACHECK(data);
  if (own_gpu_data_) {
    BASE_GPU_CHECK(cudaFree(gpu_ptr_));
  }
  gpu_ptr_ = data;
  head_ = HEAD_AT_GPU;
  own_gpu_data_ = false;
#else
  NO_GPU;
#endif
}

set_head()

void apollo::perception::base::SyncedMemory::set_head ( SyncedHead  head )

inline

在文件 syncedmem.h 第 117 行定义.

{ head_ = head; }

set_head_cpu()

void apollo::perception::base::SyncedMemory::set_head_cpu ( )

inline

在文件 syncedmem.h 第 119 行定义.

{ set_head(HEAD_AT_CPU); }

set_head_gpu()

void apollo::perception::base::SyncedMemory::set_head_gpu ( )

inline

在文件 syncedmem.h 第 118 行定义.

{ set_head(HEAD_AT_GPU); }

size()

size_t apollo::perception::base::SyncedMemory::size ( )

inline

在文件 syncedmem.h 第 120 行定义.

{ return size_; }

---

该类的文档由以下文件生成:

• modules/perception/common/base/syncedmem.h
• modules/perception/common/base/syncedmem.cc