• F
    add maximum limit for grid of index_select (#41127) · af8d2482
    FlyingQianMM 提交于
    * 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
    af8d2482
index_select_grad_kernel.cu 4.8 KB