Tensor should use PADDLE_ONLY_CPU macro
* Not use __CUDACC__ macro, because malloc in GPU memory do not have to used in .cu file. * Also, `PADDLE_ENFORCE` logic is wrong.
Showing
想要评论请 注册 或 登录
Fork自 PaddlePaddle / Paddle
* Not use __CUDACC__ macro, because malloc in GPU memory do not have to used in .cu file. * Also, `PADDLE_ENFORCE` logic is wrong.