条件编译太多如何优化

zhujiang73

类似下面这种情况，如果不想在一个函数里有这么多的条件编译，应该如何优化。

1. inline void SyncedMemory::to_cpu() {
   
2.   switch (head_) {
   
3.     case UNINITIALIZED: {
   
4.       CaffeMallocHost(&cpu_ptr_, size_, device_);
   
5.       caffe_memset(size_, 0, cpu_ptr_);
   
6.       head_ = HEAD_AT_CPU;
   
7.       own_cpu_data_ = true;
   
8.       break;
   
9.     }
   
10.     case HEAD_AT_GPU: {
    
11. #ifndef CPU_ONLY
    
12.       if (cpu_ptr_ == nullptr) {
    
13.         CaffeMallocHost(&cpu_ptr_, size_, device_);
    
14.         own_cpu_data_ = true;
    
15. #ifdef USE_GREENTEA
    
16.         CHECK_EQ(own_zero_copy_data_, false)
    
17.            << "Allocate host memory for a zero copy buffer.";
    
18. #endif
    
19.       }
    
20. 
    
21.       if (device_->backend() == Backend::BACKEND_CUDA) {
    
22. #ifdef USE_CUDA
    
23.         caffe_gpu_memcpy(size_, gpu_ptr_, cpu_ptr_);
    
24. #endif  // USE_CUDA
    
25.       } else {
    
26. #ifdef USE_GREENTEA
    
27.         viennacl::ocl::context &ctx = viennacl::ocl::get_context(
    
28.             device_->id());
    
29.         if (!own_zero_copy_data_) {
    
30.           greentea_gpu_memcpy(size_, (cl_mem) gpu_ptr_, 0, cpu_ptr_, &ctx);
    
31.         } else {
    
32.           void *mapped_ptr = clEnqueueMapBuffer(ctx.get_queue().handle().get(),
    
33.                                 (cl_mem) gpu_ptr_,
    
34.                                 true,
    
35.                                 CL_MAP_READ | CL_MAP_WRITE,
    
36.                                 0, size_, 0, NULL, NULL, NULL);
    
37.           CHECK_EQ(mapped_ptr, cpu_ptr_)
    
38.             << "Device claims it support zero copy"
    
39.             << " but failed to create correct user ptr buffer";
    
40.           clEnqueueUnmapMemObject(ctx.get_queue().handle().get(),
    
41.                                   (cl_mem) gpu_ptr_,
    
42.                                   mapped_ptr, 0, NULL, NULL);
    
43.         }
    
44.         ctx.get_queue().finish();
    
45. #endif
    
46.       }
    
47.       head_ = SYNCED;
    
48. #else
    
49.       NO_GPU;
    
50. #endif  // !CPU_ONLY
    
51.       break;
    
52.     }
    
53.     case HEAD_AT_CPU:
    
54.     case SYNCED:
    
55.       break;
    
56.   }
    
57. }
    

复制代码

hellioncu

分别拆成多个方法

action08

可以看看借鉴一下kernel的代码，汇编也用上了，



有时为了性能，工程师也是蛮拼的

zhujiang73

hellioncu 发表于 2017-02-16 08:49
分别拆成多个方法

      
      不能改变原设计接口。

      类似的代码还有很多，看了一下共有三种情况，CPU_ONLY， USE_CUDA，USE_GREENTEA  我不习惯 if 和 #ifdef 这样的交叉。

hellioncu

zhujiang73 发表于 2017-02-17 13:28
不能改变原设计接口。

      类似的代码还有很多，看了一下共有三种情况，CP ...

三种情况如果代码差别较大，那就内部分三种写，稍微冗余一些代码，这样看起来容易些

买单老

改变这样发风格，如果采用分模块的话，就要引入模块，那么在运行时不就增加开销了吗。当前至少编译器的开销，怕什么