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. 20 4月, 2023 1 次提交
    • Z
      move_elementwise_raw (#53010) · 7a72f7a2
      zhangyuqin1998 提交于
      * setup
      
      * Update elementwise_kernel.cc
      
      * Update elementwise_kernel.cc
      
      * fix
      
      * fix
      
      * Update elementwise_kernel.cu
      
      * fix
      
      * Update elementwise_kernel.cc
      
      * Update elementwise_kernel.cc
      
      * Update elementwise_kernel.cc
      
      * Update elementwise_kernel.cc
      
      * Update elementwise_kernel.cc
      
      * Update elementwise_kernel.cc
      7a72f7a2
  3. 30 3月, 2023 1 次提交