1. 28 4月, 2023 1 次提交
    • B
      Dropout optimize & clean broadcast inT and ElementwiseType (#52969) · d611e48c
      Bo Zhang 提交于
      * change judgement for DropoutGradGPUKernelDriver
      
      * add UnrollerWithoutVecSize and after this Loaddata to be refined
      
      * pass unittest
      
      * use same unroller with XPU
      
      * BroadcastWithInt64Index
      
      * BroadcastDataLoader template partial specialization
      
      * fix compile errs in ROCms
      
      * clean ElementwiseT and InT for BroadcastKernel
      
      * default axis and clean inT
      
      * remove redundant fast divmod computation
      
      * optimize drop_nd & drop_nd_grad
      
      * optimize BroadcastDataLoader bf16 fp16
      
      * rm InT etc. after merge develop
      
      * delete constexpr for windows ci
      
      * fix conflict
      
      * fix conflic with develop
      
      * fix conflic
      
      * new clean
      
      * clean
      d611e48c
  2. 21 11月, 2022 1 次提交
  3. 20 5月, 2022 1 次提交
  4. 14 4月, 2022 1 次提交
  5. 28 3月, 2022 1 次提交
  6. 18 3月, 2022 1 次提交
    • C
      [Phi] move reduce_grad kernel into phi (#40522) · 70726696
      chentianyu03 提交于
      * move reduce_mean_grad kernel into phi
      
      * move reduce_max/min_grad into phi
      
      * remove raw max/min grad kernel
      
      * fix bug
      
      * fix max/min grad error
      
      * move all reduce_grad kernel into one file
      
      * add prod grad kernel
      
      * add infermeta for prod kernel
      70726696
  7. 11 3月, 2022 1 次提交
    • C
      [Phi] Reduce grad (#40263) · f452ad5c
      chentianyu03 提交于
      * add reduce_sum grad kernel
      
      * add reduce_grad
      
      * modify reduce grad
      
      * update reduce grad functions
      
      * fix build error
      
      * add argument mapping
      
      * move cast input after grad
      
      * add dims.size=1 cpu reduce_sum grad compute method
      
      * update reduce grad GPU
      
      * remove raw reduce_sum_grad kernel
      
      * modify header files
      
      * add namespace funcs for reduce_grad_funcstions
      f452ad5c
  8. 20 2月, 2022 2 次提交
  9. 09 2月, 2022 1 次提交