Caffe解惑：为什么Caffe里头有mutable_cpu_data和cpu_data

Posted 2023-02-04 xizero00

一开始看代码的时候会纳闷，为啥caffe里头又一个 cpu_data 还要有一个mutable_cpu_data
其实从字面就可以看出来这个mutable就是易变的意思啦。
二话不说翻出源代码看看究竟有啥区别:

const void* SyncedMemory::cpu_data() 
  to_cpu();
  return (const void*)cpu_ptr_;

void* SyncedMemory::mutable_cpu_data() 
  to_cpu();
  head_ = HEAD_AT_CPU;
  return cpu_ptr_;

他们之间就相差一个 head_ = HEAD_AT_CPU; 这是为什么呢？为啥需要这样做？


我想原因就是 当你想读取数据的时候请使用cpu_data 当然想修改数据的时候请你使用mutable_cpu_data。 这样就提示系统数据我改过啦，你要小心了的意思(我只知道数据一定在CPU上)。

另外还有一个 gpu_data和mutable_gpu_data，也是这样的原因。