1. 24 12月, 2021 2 次提交
    • C
      [pten] combine reduce_cuda codes (#38328) · 08941eda
      chentianyu03 提交于
      * combine reduce_cuda codes
      
      * support float16 in pten redcue_mean
      
      * replace ReduceCudaKernel impl with pten reduce impl
      
      * mv reduce funcs into reduce_cuda_impl
      
      * rm unsed codes and headers
      
      * mv GetReduceDim into reduce_cuda_impl
      
      * recover GetReduceDim in reduce_op.h
      
      * add new dispatch macro
      
      * fix pool op output not inited and cause transform to pten::denseTensor error
      
      * fix output tensor not initialized error
      
      * rename new dispatch macro and format code style
      
      * rm reduce_functor_op.h file
      08941eda
    • Z
      [Unify Tensors PR #1] Replaced pten::Allocation with... · 42cf2bee
      Zhanlue Yang 提交于
      [Unify Tensors PR #1] Replaced pten::Allocation with shared_ptr<memory::Allocation> for Storage (#38301)
      
      * Added shared_ptr<Allocation> member & corresponding interfaces to Storage
      
      * Removed original pten::Allocation from Storage and adjusted the interfaces accordingly
      
      * Fixed issues with storage offset
      
      * Used place to malloc allocation for TensorStorage
      42cf2bee
  2. 23 12月, 2021 5 次提交
  3. 22 12月, 2021 2 次提交
  4. 21 12月, 2021 2 次提交
  5. 20 12月, 2021 1 次提交
  6. 17 12月, 2021 2 次提交
    • J
      Support multi place constructor (#38171) · 6f439e5a
      Jiabin Yang 提交于
      * support more eager tensor api
      
      * support multiple constructor for eager tensor
      
      * add place related code
      
      * polish code
      
      * specific randint with dtype of int64
      
      * Support pure cpu test
      
      * refine test in pure cpu
      
      * refine test in pure cpu
      6f439e5a
    • C
      [pten] modify reduce_sum reduce_mean args (#38216) · eaa2363e
      chentianyu03 提交于
      * modify sum mean args
      
      * add GetExpectedPtenKernelArgs for redcue_op
      
      * modify kernel args number
      
      * modify kernel args number
      eaa2363e
  7. 16 12月, 2021 2 次提交
  8. 15 12月, 2021 3 次提交
  9. 14 12月, 2021 3 次提交
  10. 13 12月, 2021 2 次提交
  11. 10 12月, 2021 1 次提交
  12. 09 12月, 2021 1 次提交
  13. 07 12月, 2021 2 次提交
  14. 06 12月, 2021 2 次提交
  15. 03 12月, 2021 3 次提交
  16. 30 11月, 2021 1 次提交
  17. 29 11月, 2021 1 次提交
    • C
      [Pten] Add reduce mean kernel, replace with mean API (#37559) · f9e9fd19
      chentianyu03 提交于
      * add pten reduce kernel
      
      * add reduce_sum kernel
      
      * update attribute args and order
      
      * make out dtype undefined
      
      * fix empty input error
      
      * merge develop branch
      
      * rename sum as reduce function
      
      * rename sum as reduce function
      
      * fix reducekernelImpl args error
      
      * add reduce cuda kernel
      
      * modify dims type to const &
      
      * remove unsed log
      
      * fix reduce_all out eigen function error
      
      * remove unused codes
      
      * add the missing sum api define and testcase
      
      * merge develop branch
      
      * fix sum test axis value error
      
      * replace pten mean kernel with reduce_mean
      
      * revcover meam cuda to original implement
      f9e9fd19
  18. 26 11月, 2021 2 次提交
  19. 25 11月, 2021 2 次提交
  20. 24 11月, 2021 1 次提交
    • Y
      elementwise_mul refactor (#37471) · c5e857d4
      YuanRisheng 提交于
      * elementwise_mul refactor
      
      * perfect code in test
      
      * delete redundant code
      
      * fix bugs when run test_multiply
      
      * adjust the location of macro
      
      * fix bugs when run ci
      c5e857d4