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

本文解析了Caffe框架中SyncedMemory类的cpu_data与mutable_cpu_data的区别,解释了这两种方法在读取和修改数据时的不同作用,并介绍了类似的GPU版本的方法。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

一开始看代码的时候会纳闷,为啥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,也是这样的原因。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值