- 28 4月, 2023 1 次提交
-
-
由 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
-
- 20 4月, 2023 1 次提交
-
-
由 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
-
- 30 3月, 2023 1 次提交
-
-
由 zhangyuqin1998 提交于
* move elementwise raw * fix * fix
-