- 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
-
- 14 4月, 2023 1 次提交
-
-
由 chenxujun 提交于
* Add digamma, dirichlet tests * Fix code
-
- 20 9月, 2022 1 次提交
-
-
由 YuanRisheng 提交于
-
- 28 7月, 2022 1 次提交
-
-
由 Xiaoxu Chen 提交于
* migrate dirichlet op kernel to phi * fix dirichlet sample memory leak
-
- 26 6月, 2022 1 次提交
-
-
由 Sing_chan 提交于
-
- 23 5月, 2022 1 次提交
-
-
由 zhouweiwei2014 提交于
* remove is_init_py of RandomGenerator, and use Global Generator if not OP seed * fix comment
-
- 17 1月, 2022 1 次提交
-
-
由 Wilber 提交于
* add pten::Place data structure. * update ci problem * fix ci problem * update * using platform::Place=pten::Place * remove BOOST_GET_CONST for CPUPlace and GPUPlace * compile pass 25%. * compile pass 45% * compile pass 60% * remove boost_get for xpu npu mlu and ipu * compile pass on cpu and gpu. * fix compile problem * fix compile error. * update * fix ci problem * update * ci approve * fix ci problem * fix ci eager test problem * remove BOOST_GET_CONST * fix npu compile
-
- 30 12月, 2021 1 次提交
-
-
由 Xiaoxu Chen 提交于
* add dirichlet sample op and cpu backend kernel * add Dirichlet op cuda kernel (#6) * add dirichlet op hip kernel Co-authored-by: NFeiyu Chan <chenfeiyu@baidu.com>
-