Fix

Author: co63oc

paddle/phi/kernels/fill_diagonal_tensor_kernel.h

@@ -39,7 +39,7 @@ DenseTensor FillDiagonalTensor(const Context& dev_ctx,
  MetaTensor meta_out(&dense_out);
  FillDiagonalTensorInferMeta(x, y, offset, dim1, dim2, &meta_out);
  FillDiagonalTensorKernel<T, Context>(
- ctx, x, y, offset, dim1, dim2, &dense_out);
+ dev_ctx, x, y, offset, dim1, dim2, &dense_out);
  return dense_out;
 }
 

paddle/phi/kernels/slice_kernel.h

@@ -66,8 +66,14 @@ DenseTensor Slice(const Context& dev_ctx,
  std::vector<int64_t> decrease_axis = {};
  SliceRawInferMeta(
  input, axes, starts, ends, infer_flags, decrease_axis, &meta_out);
- SliceKernel<T, Context>(
- ctx, input, axes, starts, ends, infer_flags, decrease_axis, &dense_out);
+ SliceKernel<T, Context>(dev_ctx,
+ input,
+ axes,
+ starts,
+ ends,
+ infer_flags,
+ decrease_axis,
+ &dense_out);
  return dense_out;
 }
 
@@ -85,7 +91,7 @@ void Slice(const Context& dev_ctx,
  input, axes, starts, ends, infer_flags, decrease_axis, &meta_out);
  if (input.initialized()) {
  SliceKernel<T, Context>(
- ctx, input, axes, starts, ends, infer_flags, decrease_axis, out);
+ dev_ctx, input, axes, starts, ends, infer_flags, decrease_axis, out);
  }
 }
 

paddle/phi/kernels/triangular_solve_kernel.h

@@ -39,7 +39,7 @@ DenseTensor TriangularSolve(const Context& dev_ctx,
  MetaTensor meta_out(&dense_out);
  TriangularSolveInferMeta(x, y, upper, transpose, unitriangular, &meta_out);
  TriangularSolveKernel<T, Context>(
- ctx, x, y, upper, transpose, unitriangular, &dense_out);
+ dev_ctx, x, y, upper, transpose, unitriangular, &dense_out);
  return dense_out;
 }