Created by: wojtuss

This patch fixes the Ernie crash reported in the issue https://github.com/PaddlePaddle/Paddle/issues/24394.

test=develop