CAFFE源码学习笔记之四-device_alternate
来源:互联网 发布:数据库版本几位数 编辑:程序博客网 时间:2024/05/27 21:49
一、前言
common中包含的头文件中有一个device_alternate,里面主要是和cuda有关系的宏:
CUDA_CHECK;
CUBLAS_CHECK;
CURAND_CHECK;
CUDA_KERNEL_LOOP;
同时还对block中的线程数以及block的维度声明为常量。
二、源码分析
首先是NO_GPU宏,主要是当设置为CPU_ONLY后,如果你仍然固执地要用forward的gpu版,那么他就会告诉你不要犯傻好么。LOG(FATAL)是glog的宏,后面集中总结一下。
#define NO_GPU LOG(FATAL) << "Cannot use GPU in CPU-only Caffe: check mode."
cudaError_t是cuda 运行时API,任何cuda API运行都有可能发生错误,只有当正常运行后才返回cudaSuccess。为什么专门写宏呢,为了避免cudaError_t重复定义,同时也是省事。
#define CUDA_CHECK(condition) \ /* Code block avoids redefinition of cudaError_t error */ \ do { \ cudaError_t error = condition; \ CHECK_EQ(error, cudaSuccess) << " " << cudaGetErrorString(error); \ } while (0)
const定义block内的线程个数为512,而block的个数表示为数据点总数除以每个block内的线程数,就是每个线程负责一个计算。
// CUDA: use 512 threads per blockconst int CAFFE_CUDA_NUM_THREADS = 512;// CUDA: number of blocks for threads.inline int CAFFE_GET_BLOCKS(const int N) { return (N + CAFFE_CUDA_NUM_THREADS - 1) / CAFFE_CUDA_NUM_THREADS;
循环以线程总数为步长,每次循环保证所有线程同时进行计算。
// CUDA: grid stride looping#define CUDA_KERNEL_LOOP(i, n) \ for (int i = blockIdx.x * blockDim.x + threadIdx.x; \ i < (n); \ i += blockDim.x * gridDim.x)
四、总结
主要是写宏可以很省事。至于cuda再开一系列。。。
0 0
- CAFFE源码学习笔记之四-device_alternate
- CAFFE源码学习笔记之五-internal_thread
- CAFFE源码学习笔记之六-Blob
- CAFFE源码学习笔记之九-data_transformer
- CAFFE源码学习笔记之十-data_layer
- CAFFE源码学习笔记之池化层pooling_layer
- CAFFE源码学习笔记之激活层
- CAFFE源码学习笔记之softmax_layer
- CAFFE源码学习笔记之batch_norm_layer
- CAFFE源码学习笔记之初始化Filler
- caffe源码c++学习笔记
- CAFFE源码学习笔记之七-layer_factory以及layer基类
- CAFFE源码学习笔记之八-阻塞队列blocking_queue
- CAFFE源码学习笔记之十一-卷积层conv_layer
- CAFFE源码学习笔记之内积层-inner_product_layer
- caffe源码学习(四) layer
- caffe 学习笔记之caffe.proto
- caffe 学习笔记之caffe.proto注释
- 【NOI2017模拟3.30】轮回(根号算法,暴力)
- 设置状态通知栏的颜色
- android中popupwindow的简单使用
- Laravel管道技术
- 《道德经》第十五章
- CAFFE源码学习笔记之四-device_alternate
- Win7如何搜索文件里包含的文字
- 使用@Controller注解为什么要配置<mvc:annotation-driven />
- 重磅教程!帮你全面彻底搞定MATERIAL DESIGN的学习笔记
- LEMP——安装配置
- 【Shiro】Apache Shiro架构之实际运用(整合到Spring中)
- Ubuntu下gcc多版本共存和版本切换
- Unger平滑
- android studio下面如何打包jar包