Skip to content

Commit faaa7d0

Browse files
committed
Add index check for index_sample gpu kernel
1 parent e01b0f6 commit faaa7d0

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

paddle/phi/kernels/gpu/index_sample_kernel.cu

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,12 @@ __global__ void IndexSampleForward(const SampleIndexT* index,
4848
ElementIndexT index_idx = index_j * index_length + index_i;
4949
ElementIndexT in_idx = index_j * input_length + index_i;
5050
SampleIndexT sample_idx = index[index_idx];
51+
PADDLE_ENFORCE(sample_idx >= 0 && sample_idx < input_length,
52+
"Variable value (index) of OP(index_sample) "
53+
"expected >= 0 and < %ld, but got %ld. Please check input "
54+
"value.",
55+
input_length,
56+
sample_idx);
5157
out_data[index_idx] = in_data[in_idx - index_i + sample_idx];
5258
}
5359
}

0 commit comments

Comments
 (0)