1. 10 2月, 2022 1 次提交
    • 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