1. 05 1月, 2022 1 次提交
  2. 04 1月, 2022 2 次提交
  3. 29 12月, 2021 2 次提交
  4. 28 12月, 2021 3 次提交
  5. 27 12月, 2021 2 次提交
    • L
      Support multi-outputs feature for broadcast ops (#38329) · 89d38f55
      limingshu 提交于
      * No harm to KP
      
      * Pass the compile stage
      
      * change the WriteData function
      
      * fix template bugs and pass ctest of current elementwise
      
      * for passing partial template specialization of tempalte function in CI-ROCm
      
      * To make 'WriteData' funtion flexible.
      
      * a less harmful way to support multi-output
      
      * a less harmful way to support multi-output
      89d38f55
    • C
      [PTen] Move cast kernel impl (#38382) · 1fb734d7
      Chen Weihang 提交于
      * rename to api to copy_to
      
      * revert needless change
      
      * polish format
      1fb734d7
  6. 26 12月, 2021 1 次提交
    • C
      [PTen] Move copy kernel impl (#38421) · 73819658
      Chen Weihang 提交于
      * add register general kernel marco
      
      * move copy kernel impl
      
      * revert needless change
      
      * polish details
      
      * fix xpu compil faild
      
      * fix xpu compile failed
      
      * polish format
      73819658
  7. 24 12月, 2021 1 次提交
    • 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
  8. 23 12月, 2021 2 次提交
  9. 21 12月, 2021 2 次提交
  10. 20 12月, 2021 3 次提交
  11. 17 12月, 2021 1 次提交
  12. 16 12月, 2021 3 次提交
  13. 15 12月, 2021 2 次提交
  14. 13 12月, 2021 1 次提交
  15. 10 12月, 2021 2 次提交
  16. 09 12月, 2021 1 次提交