Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
未验证 提交 2e76d5ad 编写于 作者: C chentianyu03 提交者： GitHub
浏览文件
操作

pylayer support HIP (#38184)

上级 af30f545
隐藏空白更改
内联 并排
Showing with 2 addition and 2 deletion
paddle/fluid/operators/py_layer_op.cc
浏览文件 @ 2e76d5ad
...@@ -220,7 +220,7 @@ REGISTER_OP_CPU_KERNEL( ...@@ -220,7 +220,7 @@ REGISTER_OP_CPU_KERNEL(
::paddle::platform::complex<float>>, ::paddle::platform::complex<float>>,
ops::PyLayerOpKernel<paddle::platform::CPUDeviceContext, ops::PyLayerOpKernel<paddle::platform::CPUDeviceContext,
::paddle::platform::complex<double>>); ::paddle::platform::complex<double>>);
#ifdef PADDLE_WITH_CUDA #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
REGISTER_OP_CUDA_KERNEL( REGISTER_OP_CUDA_KERNEL(
py_layer, ops::PyLayerOpKernel<paddle::platform::CUDADeviceContext, float>, py_layer, ops::PyLayerOpKernel<paddle::platform::CUDADeviceContext, float>,
ops::PyLayerOpKernel<paddle::platform::CUDADeviceContext, ops::PyLayerOpKernel<paddle::platform::CUDADeviceContext,
...@@ -239,4 +239,4 @@ REGISTER_OP_CUDA_KERNEL( ...@@ -239,4 +239,4 @@ REGISTER_OP_CUDA_KERNEL(
::paddle::platform::complex<float>>, ::paddle::platform::complex<float>>,
ops::PyLayerOpKernel<paddle::platform::CUDADeviceContext, ops::PyLayerOpKernel<paddle::platform::CUDADeviceContext,
::paddle::platform::complex<double>>); ::paddle::platform::complex<double>>);
#endif // PADDLE_WITH_CUDA #endif // PADDLE_WITH_CUDA || PADDLE_WITH_HIP
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或