• J
    Optimize where_index_op(prefix sum) (#30601) · 6ec4e640
    jiangcheng 提交于
    * new optimize for where_index_op with prefix sum version.
    
    * write a scan prefix sum kernel with stream for where index op.
    
    * optimize where_index by using cub::DeviceScan::InclusiveSum instead of imperfect self-kernel.
    
    * remove CheckTrue struct and rename stide_array for readable.
    
    * optimize variable name for readable.
    
    * optimize function name and annotation.
    6ec4e640
where_index_op.cu 6.5 KB