slice-check

Author: zhanghonggeng

paddle/phi/kernels/funcs/radix_sort.cu

@@ -18,8 +18,8 @@
 namespace phi {
 namespace funcs {
 
+#ifdef PADDLE_WITH_CUDA
 namespace {
-
 template <typename T>
 struct CudaType {
  using type = T;
@@ -113,5 +113,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