提交 05dc78ea 编写于 作者: Y Yu Yang

Merge branch 'master' of github.com:baidu/Paddle

......@@ -16,6 +16,8 @@ limitations under the License. */
#ifndef HL_DEVICE_FUNCTIONS_CUH_
#define HL_DEVICE_FUNCTIONS_CUH_
#if defined(__CUDA_ARCH__) && __CUDA_ARCH__ < 600
namespace hppl {
static __inline__ __device__ double atomicAdd(double* address, double val) {
......@@ -38,4 +40,6 @@ static __inline__ __device__ double atomicAdd(double* address, double val) {
using hppl::atomicAdd;
#endif
#endif /* HL_DEVICE_FUNCTIONS_CUH_ */
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册