Created by: typhoonzero

Provide another fix to https://github.com/PaddlePaddle/Paddle/pull/10626#pullrequestreview-120104317