We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 97a9e5f commit 2650cc8Copy full SHA for 2650cc8
paddle/phi/kernels/selected_rows/shape_kernel.cc
@@ -95,7 +95,7 @@ PD_REGISTER_KERNEL(shape_sr,
95
}
96
#endif
97
98
-#ifdef PADDLE_WITH_CUSTOM_DEVICE
+#if defined(PADDLE_WITH_CUSTOM_DEVICE) && !defined(PADDLE_WITH_CUDA)
99
PD_REGISTER_KERNEL(shape_sr,
100
Custom,
101
ALL_LAYOUT,
@@ -171,7 +171,7 @@ PD_REGISTER_KERNEL(shape64_sr,
171
172
173
174
175
PD_REGISTER_KERNEL(shape64_sr,
176
177
0 commit comments