error: function "atomicAdd(double *, double)" has already been defined

最新推荐文章于 2021-06-18 12:22:24 发布

原创最新推荐文章于 2021-06-18 12:22:24 发布 · 2.1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#deeplab

deeplab 专栏收录该内容

1 篇文章

订阅专栏

本文解决了在使用CUDA8.0编译DeepLab_v2时遇到的原子操作定义冲突问题。通过修改common.cuh文件中的原子加操作实现，确保了与CUDA8.0的兼容性。

在编译deeplab_V2的时候，用Cuda-8.0，出现上述错误。
之前在cuda-7.5的时候，没有出现这个错误。在网上找到错误的原因，记录下来。
原文地址

CUDA 8.0 provides a definition of atomicAdd on double quantities that was not present in previous CUDA toolkits. The code you are working with also apparently provides its own definition/implementation, and this is the source of the error message. The correct fix is to make source code changes to the software in question to make it compatible with CUDA 8.

I had to modify the file common.cuh from the DeepLab_v2 master branch in the following way:

#ifndef CAFFE_COMMON_CUH_
#define CAFFE_COMMON_CUH_

#include <cuda.h>

  #if !defined(__CUDA_ARCH__) || __CUDA_ARCH__ >= 600

  #else
  static __inline__ __device__ double atomicAdd(double *address, double val) {
    unsigned long long int* address_as_ull = (unsigned long long int*)address;
    unsigned long long int old = *address_as_ull, assumed;
    if (val==0.0)
      return __longlong_as_double(old);
    do {
      assumed = old;
      old = atomicCAS(address_as_ull, assumed, __double_as_longlong(val +__longlong_as_double(assumed)));
    } while (assumed != old);
    return __longlong_as_double(old);
  }


  #endif
#endif