1. 11 2月, 2022 1 次提交
  2. 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
  3. 21 1月, 2022 1 次提交
  4. 18 1月, 2022 1 次提交
  5. 17 1月, 2022 1 次提交
  6. 12 1月, 2022 2 次提交
  7. 11 1月, 2022 2 次提交
  8. 05 1月, 2022 2 次提交
    • L
      optimize elementwise_mul_grad using new interfaces (#37728) · 36a102f8
      Lijunhui 提交于
      * init commit: new elem_mul_grad
      
      * add template speciallization for complex in multiply
      
      * reply review comments
      
      * correct dx and dy computation when T is complex
      
      * reply review comments
      
      * update to new ReduceRunctor
      
      * mul-output broadcast
      
      * call functions
      
      * call functions with comments
      
      * remove comments
      36a102f8
    • C
      implementation of broadcast div backward by reduce (#38044) · 55cd9cb8
      crystal 提交于
      * add elementwise div
      
      * move mul and div grad functor
      
      * Combine multiple CUDA kernels
      
      * Update the reduce interface call
      
      * add multi-output
      
      * add multi-output div
      
      * add branch judge
      
      * Package branch
      
      * Combine the x and y functions into one
      55cd9cb8
  9. 04 1月, 2022 1 次提交
  10. 16 12月, 2021 1 次提交
  11. 15 9月, 2021 1 次提交