未验证 提交 3b4dcad7 编写于 作者: Q Qi Li 提交者: GitHub

[ROCM] update PADDLE_WITH_ROCM to PADDLE_WITH_HIP, test=develop (#32487)

上级 2f351ed5
...@@ -147,7 +147,7 @@ REGISTER_OP_CPU_KERNEL( ...@@ -147,7 +147,7 @@ REGISTER_OP_CPU_KERNEL(
ops::ExpandAsGradKernel<paddle::platform::CPUDeviceContext, int64_t>, ops::ExpandAsGradKernel<paddle::platform::CPUDeviceContext, int64_t>,
ops::ExpandAsGradKernel<paddle::platform::CPUDeviceContext, float>, ops::ExpandAsGradKernel<paddle::platform::CPUDeviceContext, float>,
ops::ExpandAsGradKernel<paddle::platform::CPUDeviceContext, double>); ops::ExpandAsGradKernel<paddle::platform::CPUDeviceContext, double>);
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_ROCM) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
REGISTER_OP_CUDA_KERNEL( REGISTER_OP_CUDA_KERNEL(
expand_as, ops::ExpandAsKernel<paddle::platform::CUDADeviceContext, float>, expand_as, ops::ExpandAsKernel<paddle::platform::CUDADeviceContext, float>,
ops::ExpandAsKernel<paddle::platform::CUDADeviceContext, double>, ops::ExpandAsKernel<paddle::platform::CUDADeviceContext, double>,
......
...@@ -129,7 +129,7 @@ REGISTER_OP_CPU_KERNEL( ...@@ -129,7 +129,7 @@ REGISTER_OP_CPU_KERNEL(
ops::ExpandAsV2GradKernel<paddle::platform::CPUDeviceContext, int64_t>, ops::ExpandAsV2GradKernel<paddle::platform::CPUDeviceContext, int64_t>,
ops::ExpandAsV2GradKernel<paddle::platform::CPUDeviceContext, float>, ops::ExpandAsV2GradKernel<paddle::platform::CPUDeviceContext, float>,
ops::ExpandAsV2GradKernel<paddle::platform::CPUDeviceContext, double>); ops::ExpandAsV2GradKernel<paddle::platform::CPUDeviceContext, double>);
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_ROCM) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
REGISTER_OP_CUDA_KERNEL( REGISTER_OP_CUDA_KERNEL(
expand_as_v2, expand_as_v2,
ops::ExpandAsV2Kernel<paddle::platform::CUDADeviceContext, float>, ops::ExpandAsV2Kernel<paddle::platform::CUDADeviceContext, float>,
......
...@@ -273,7 +273,7 @@ REGISTER_OP_CPU_KERNEL( ...@@ -273,7 +273,7 @@ REGISTER_OP_CPU_KERNEL(
ops::ExpandGradKernel<paddle::platform::CPUDeviceContext, double>, ops::ExpandGradKernel<paddle::platform::CPUDeviceContext, double>,
ops::ExpandGradKernel<paddle::platform::CPUDeviceContext, int>, ops::ExpandGradKernel<paddle::platform::CPUDeviceContext, int>,
ops::ExpandGradKernel<paddle::platform::CPUDeviceContext, int64_t>); ops::ExpandGradKernel<paddle::platform::CPUDeviceContext, int64_t>);
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_ROCM) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
REGISTER_OP_CUDA_KERNEL( REGISTER_OP_CUDA_KERNEL(
expand, ops::ExpandKernel<paddle::platform::CUDADeviceContext, float>, expand, ops::ExpandKernel<paddle::platform::CUDADeviceContext, float>,
ops::ExpandKernel<paddle::platform::CUDADeviceContext, double>, ops::ExpandKernel<paddle::platform::CUDADeviceContext, double>,
......
...@@ -278,7 +278,7 @@ REGISTER_OP_CPU_KERNEL( ...@@ -278,7 +278,7 @@ REGISTER_OP_CPU_KERNEL(
ops::ExpandV2GradKernel<paddle::platform::CPUDeviceContext, double>, ops::ExpandV2GradKernel<paddle::platform::CPUDeviceContext, double>,
ops::ExpandV2GradKernel<paddle::platform::CPUDeviceContext, int>, ops::ExpandV2GradKernel<paddle::platform::CPUDeviceContext, int>,
ops::ExpandV2GradKernel<paddle::platform::CPUDeviceContext, int64_t>); ops::ExpandV2GradKernel<paddle::platform::CPUDeviceContext, int64_t>);
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_ROCM) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
REGISTER_OP_CUDA_KERNEL( REGISTER_OP_CUDA_KERNEL(
expand_v2, ops::ExpandV2Kernel<paddle::platform::CUDADeviceContext, float>, expand_v2, ops::ExpandV2Kernel<paddle::platform::CUDADeviceContext, float>,
ops::ExpandV2Kernel<paddle::platform::CUDADeviceContext, double>, ops::ExpandV2Kernel<paddle::platform::CUDADeviceContext, double>,
......
...@@ -157,7 +157,7 @@ REGISTER_OP_CPU_KERNEL( ...@@ -157,7 +157,7 @@ REGISTER_OP_CPU_KERNEL(
ops::MeshgridGradKernel<paddle::platform::CPUDeviceContext, int64_t>, ops::MeshgridGradKernel<paddle::platform::CPUDeviceContext, int64_t>,
ops::MeshgridGradKernel<paddle::platform::CPUDeviceContext, int>, ops::MeshgridGradKernel<paddle::platform::CPUDeviceContext, int>,
ops::MeshgridGradKernel<paddle::platform::CPUDeviceContext, double>); ops::MeshgridGradKernel<paddle::platform::CPUDeviceContext, double>);
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_ROCM) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
REGISTER_OP_CUDA_KERNEL( REGISTER_OP_CUDA_KERNEL(
meshgrid, ops::MeshgridKernel<paddle::platform::CUDADeviceContext, float>, meshgrid, ops::MeshgridKernel<paddle::platform::CUDADeviceContext, float>,
ops::MeshgridKernel<paddle::platform::CUDADeviceContext, double>, ops::MeshgridKernel<paddle::platform::CUDADeviceContext, double>,
......
...@@ -286,7 +286,7 @@ REGISTER_OP_CPU_KERNEL( ...@@ -286,7 +286,7 @@ REGISTER_OP_CPU_KERNEL(
ops::TileGradKernel<paddle::platform::CPUDeviceContext, double>, ops::TileGradKernel<paddle::platform::CPUDeviceContext, double>,
ops::TileGradKernel<paddle::platform::CPUDeviceContext, int>, ops::TileGradKernel<paddle::platform::CPUDeviceContext, int>,
ops::TileGradKernel<paddle::platform::CPUDeviceContext, int64_t>); ops::TileGradKernel<paddle::platform::CPUDeviceContext, int64_t>);
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_ROCM) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
REGISTER_OP_CUDA_KERNEL( REGISTER_OP_CUDA_KERNEL(
tile, ops::TileKernel<paddle::platform::CUDADeviceContext, float>, tile, ops::TileKernel<paddle::platform::CUDADeviceContext, float>,
ops::TileKernel<paddle::platform::CUDADeviceContext, double>, ops::TileKernel<paddle::platform::CUDADeviceContext, double>,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册