Skip to content

Commit 6e1cf93

Browse files
fix data is nullptr (#73526)
1 parent 0f586c4 commit 6e1cf93

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paddle/phi/kernels/gpu/masked_fill_grad_kernel.cu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -243,12 +243,12 @@ void GPUMaskedFillGrad(const phi::GPUContext& dev_ctx,
243243

244244
int vec_size = 4;
245245
vec_size = std::min(phi::GetVectorizedSize(out_grad_data), vec_size);
246-
if (x_grad) {
246+
if (x_grad && x_grad->initialized()) {
247247
x_grad_data = x_grad->data<T>();
248248
vec_size = std::min(phi::GetVectorizedSize(x_grad_data), vec_size);
249249
}
250250

251-
if (value_grad) {
251+
if (value_grad && value_grad->initialized()) {
252252
value_grad_data = value_grad->data<T>();
253253
vec_size = std::min(phi::GetVectorizedSize(value_grad_data), vec_size);
254254
}

0 commit comments

Comments
 (0)