Created by: zhaoyuchen2018

The op should handle k=1024 Fix seq_len < warpsize error.

test=develop

Signed-off-by: zhaoyuchen zhaoyuchen01@baidu.com