1. 13 2月, 2022 1 次提交
  2. 11 2月, 2022 2 次提交
  3. 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
  4. 09 2月, 2022 6 次提交
  5. 08 2月, 2022 3 次提交
  6. 06 2月, 2022 1 次提交
  7. 29 1月, 2022 1 次提交
    • C
      [PTen] Tidy pten core headers (#39188) · dd990981
      Chen Weihang 提交于
      * open header for custom kernel
      
      * add core utils
      
      * tidy core code
      
      * tify header
      
      * tidy include
      
      * tidy namespace
      
      * resolve conflit
      
      * fix unittest and coverage
      
      * remove platform using
      
      * resolve conflict
      
      * resolve conflict
      
      * fix digamma namespace error
      
      * fix xpu full kernel error
      
      * fix xpu full kernel error
      
      * polish details
      
      * add place for lib storage
      dd990981
  8. 28 1月, 2022 2 次提交
  9. 27 1月, 2022 4 次提交
    • Z
      implement AllocateFrom (#39280) · d89f246c
      zhangkaihuo 提交于
      d89f246c
    • Y
      refactor elementwise sub grad (#39225) · 7a1e1193
      YuanRisheng 提交于
      7a1e1193
    • A
      [PTen]Support AllocateFrom in Tensor and Alloc/HostAlloc in Context (#39022) · 5631da9c
      Aurelius84 提交于
      * Support allocate_from in Tensor and allocate_data in Context
      
      * fix #ifdef CUDA
      
      * fix cycle depends
      
      * fix test_xxx_dev_api failed
      
      * fix windows compiling error
      
      * fix unittest
      
      * modify into PImpl
      
      * fix selected rows
      
      * add TODO comment
      
      * refine interface according reviewer
      5631da9c
    • Z
      Add SparseCooTensor and SparseCsrTensor (#38906) · a7edb3f3
      zhangkaihuo 提交于
      * fix bug:
      1. atten: set the default value of attn_dropout_rate to None
      2. ffn: add activation parameter
      
      * for pure fp16
      
      * Add a SparseCsrTensor
      
      * remove unused functional
      
      * remove const
      
      * remove SetMemoberTensor
      
      * remove non_zero_nums_, the number of non zero elements of each batch can be obtained from the crows
      
      * SparseCooTensor
      
      * add SetMember
      
      * merge upstream; add SetMember
      
      * merge upstream
      
      * merge upstream; add newline at end of file
      
      * add newline at end of file
      
      * remove newline at end of file
      
      * remove newline at end of file
      
      * stash
      
      * user pten::framework::make_ddim
      
      * user pten::framework::make_ddim
      
      * merge upstream; use the latest mutable_data
      
      * merge upstream; use the latest mutable_data
      
      * return mutable dense tensor
      a7edb3f3
  10. 26 1月, 2022 1 次提交
  11. 25 1月, 2022 3 次提交
  12. 24 1月, 2022 2 次提交
    • Y
      [Pten]Refactor elementwise_add grad / double grad / triple grad Kernel and... · 3bf3a6ee
      YuanRisheng 提交于
      [Pten]Refactor elementwise_add grad / double grad / triple grad Kernel and move them to pten (#39048)
      
      * refactor elementwise add grad
      
      * fix compile bugs
      
      * fix unit test bugs
      
      * fix file conflicts
      
      * fix bugs when buildPtenContext
      3bf3a6ee
    • [Refactoring Tensor PR #5] replace storage with pten allocation (#39085) · a56e16a7
      石晓伟 提交于
      * updates callers, test=develop
      
      * updates tensor, test=develop
      
      * fixes errors, test=develop
      
      * remove some dtypes, test=develop
      
      * fix errors in the base storage modification, test=develop
      
      * fixes a bug, test=develop
      
      * fixes the bugs in push the whole, test=develop
      
      * updates, test=develop
      
      * update
      
      * update, test=develop
      
      * fixes the mac-py3 CI, test=develop
      
      * remove the storage impl, test=develop
      
      * updates some codes, test=develop
      
      * update, test=develop
      
      * updates pten allocation, test=develop
      a56e16a7
  13. 22 1月, 2022 1 次提交
  14. 21 1月, 2022 5 次提交
  15. 20 1月, 2022 2 次提交
  16. 18 1月, 2022 2 次提交
  17. 17 1月, 2022 1 次提交
    • W
      [Pten] Replace platform::Place to pten::Place. (#38899) · c48a9ad5
      Wilber 提交于
      * add pten::Place data structure.
      
      * update ci problem
      
      * fix ci problem
      
      * update
      
      * using platform::Place=pten::Place
      
      * remove BOOST_GET_CONST for CPUPlace and GPUPlace
      
      * compile pass 25%.
      
      * compile pass 45%
      
      * compile pass 60%
      
      * remove boost_get for xpu npu mlu and ipu
      
      * compile pass on cpu and gpu.
      
      * fix compile problem
      
      * fix compile error.
      
      * update
      
      * fix ci problem
      
      * update
      
      * ci approve
      
      * fix ci problem
      
      * fix ci eager test problem
      
      * remove BOOST_GET_CONST
      
      * fix npu compile
      c48a9ad5
  18. 15 1月, 2022 1 次提交
  19. 14 1月, 2022 1 次提交