add maximum limit for grid of index_select (#41127)
* limit grid dim for index select * mv LimitGridDim into gpu_launch_config.h * fix conflicts * fix conflicts * fix code style * set block to 256 * fix grid setting * set dtype of block_dim to unsigned int
Showing
想要评论请 注册 或 登录