Created by: zchen0211

Fix https://github.com/PaddlePaddle/Paddle/issues/4167

Will add GPU when gpu parts are ready.