• H
    move Masked select to pten (#39193) · e2ad433b
    hong 提交于
    * move masked select cpu kernel
    
    * add masked selected gpu kernel; test=develop
    
    * fix bugs; test=develop
    
    * bug fix; test=develop
    
    * bug fix; test=develop
    
    * add namespace to set mask array; test=develop
    
    * fix bug; test=develop
    
    * fix bugs; test=develop
    
    * fix ddim bug; test=develop
    
    * fix npu op bug; test=develop
    
    * fix xpu dependecy bug; test=develop
    
    * move kernel args to sig.cc; test=develop
    e2ad433b
masked_select_kernel.cu 4.2 KB