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 8d34517 commit 76ba052Copy full SHA for 76ba052
paddle/phi/kernels/funcs/index_impl.cu.h
@@ -54,7 +54,7 @@ __global__ void VectorizedIndexKernel(T *out,
54
template <typename T, typename Functor>
55
void IndexKernel(const KPDevice &dev_ctx, DenseTensor *out, Functor func) {
56
int numel = out->numel();
57
- T *out_data = out->mutable_data<T>(dev_ctx.GetPlace());
+ T *out_data = dev_ctx.template Alloc<T>(out);
58
if (numel <= 0) return;
59
int vec_size = phi::GetVectorizedSize(out_data);
60
#ifdef PADDLE_WITH_XPU_KP
0 commit comments