• Add topk op (#3760) · 3fbb692d
    武毅 提交于
    * init add
    
    * add topk op
    
    * someupdate
    
    * fix style check
    
    * add test py file
    
    * update top k cuda kernel
    
    * follow comments
    
    * remove debug print
    
    * fix casting error
    
    * fix casting error
    
    * fix casting error
    
    * fix rename bug...
    
    * fix travis
    3fbb692d
top_k_op.cu 9.6 KB