slice-check

Author: zhanghonggeng

paddle/phi/kernels/funcs/radix_sort.cu

@@ -19,6 +19,7 @@ namespace phi {
 namespace funcs {
 
 namespace {
+#ifdef PADDLE_WITH_CUDA
 
 template <typename T>
 struct CudaType {
@@ -113,5 +114,6 @@ INSTANTIATE_SORT_PAIRS(int64_t, 4)
 INSTANTIATE_SORT_PAIRS(int32_t, 8)
 INSTANTIATE_SORT_PAIRS(int64_t, 8)
 
+#endif
 }  // namespace funcs
 }  // namespace phi

paddle/phi/kernels/funcs/radix_sort.h

@@ -13,13 +13,16 @@
 // limitations under the License.
 
 #pragma once
+#ifdef PADDLE_WITH_CUDA
 #include <cub/cub.cuh>
+#endif
 #include "paddle/phi/backends/gpu/gpu_context.h"
 #include "paddle/phi/core/dense_tensor.h"
 
 namespace phi {
 namespace funcs {
 
+#ifdef PADDLE_WITH_CUDA
 template <int kValueSize>
 struct OpaqueTypeRadix {
   uint8_t data[kValueSize];
@@ -76,5 +79,6 @@ void RadixSortPairs(const phi::GPUContext& dev_ctx,
       end_bit);
 }
 
+#endif
 }  // namespace funcs
 }  // namespace phi

paddle/phi/kernels/funcs/stride_utils.h

@@ -544,7 +544,7 @@ static inline void ScatterAddStride(
   *numel = num;
 }
 
-#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
+#if defined(PADDLE_WITH_CUDA)
 
 static inline std::vector<phi::DenseTensor> expandTensors(
     const phi::GPUContext& dev_ctx,

paddle/phi/kernels/gpu/index_elementwise_get_grad_kernel.cu

@@ -401,6 +401,7 @@ void IndexElementwiseGetGradKernel(const Context& dev_ctx,
                         phi::DataType::INT64));
 
   if (accumulate && index.size() == 1 && !is_combined) {
+#ifdef PADDLE_WITH_CUDA
     IndexPutWithSortKernel<T, int64_t>(dev_ctx,
                                        x,
                                        out_grad,
@@ -412,19 +413,21 @@ void IndexElementwiseGetGradKernel(const Context& dev_ctx,
                                        slice_offset,
                                        accumulate,
                                        x_grad);
-  } else {
-    GPUIndexElementwiseGetGrad<T, int64_t>(dev_ctx,
-                                           x,
-                                           out_grad,
-                                           index,
-                                           input_dims,
-                                           input_strides,
-                                           index_dims,
-                                           index_strides,
-                                           slice_offset,
-                                           accumulate,
-                                           x_grad);
+    return;
+#endif
   }
+
+  GPUIndexElementwiseGetGrad<T, int64_t>(dev_ctx,
+                                         x,
+                                         out_grad,
+                                         index,
+                                         input_dims,
+                                         input_strides,
+                                         index_dims,
+                                         index_strides,
+                                         slice_offset,
+                                         accumulate,
+                                         x_grad);
 }
 
 }  // namespace phi