- 08 2月, 2022 17 次提交
-
-
由 hong 提交于
* find gpu kernel in pten factory; test=develop * check in functional kernel first; test=develop
-
由 ziyoujiyi 提交于
* delete gloo connect retry * the_one_ps dirs reconstruct * . * . * create the_one_ps dirs * create the_one_ps dirs * create the_one_ps dirs * create the_one_ps dirs * create the_one_ps dirs * create the_one_ps dirs * the one ps dirs modify * the one ps dirs modify * the one ps dirs modify * the one ps dirs modify * refactor ps optimize * refactor ps optimize * refactor ps optimize * . * . * . * . * . * . * refactor theoneps * the_one_ps * add ps pass unittest * add ps pass unittest * ps unitest frame * ps unittest frame * ps unittest frame * ps unittest frame * ps unittest frame * ps unittest frame * ps unittest frame * ps unittest frame * ps unittest frame * ps unittest frame * ps unittest frame * ps unittest frame * ps unittest frame * ps unittest frame * ps unittest frame * ps unittest frame * ps unittest frame * add cpu_async_ps_mode test * add cpu_async_ps_mode test * add cpu_async_ps_mode test * ps unittest ready * ps unittest ready * solve dist_pass init conflict * solve import CommContext error * unittest ok * implement AllocateFrom * solve setup.py.in conflict * solve conflict * solve conflict * solve conflict * . * . Co-authored-by: Nzkh2016 <zhangkaihuo@baidu.com> -
由 zhangbo9674 提交于
* add concat & split * add concat kernel * add concat unittest * add split unittest
-
由 Wilber 提交于
* gpu_context.. * update * update * update
-
由 wanghuancoder 提交于
-
由 yaoxuefeng 提交于
add downpour_ctr_accessor
-
由 yaoxuefeng 提交于
-
由 niuliling123 提交于
* Replace clip, bce_loss, full and full_like with elementwise
-
由 Baibaifan 提交于
-
由 Chen Weihang 提交于
* adapt selectedrows in execution * impl selected rows branch * support selectedrow in infershape utils * fix device compile failed * fix new exe test failed * revert some changes
-
由 From00 提交于
* Rough implementation for experiment * Support allocate cuda managed memory * Fix CI error * Modify UT * Check whether support memory oversubscription * Fix ROCM Compile error * Fix ROCM Compile error * Fix UT cuda_managed_memory_test * Set UT timeout to 40 * Add UT OOMExceptionTest * Set UT timeout to 50
-
由 Chen Weihang 提交于
* fix pten reduce dispatch bug * add cast beforce reduce * fix test failed
-
由 Leo Chen 提交于
-
由 Leo Chen 提交于
-
由 Yan Chunwei 提交于
-
由 Zhanlue Yang 提交于
-
由 sneaxiy 提交于
* hack custom op * add ut * skip windows ci
-
- 07 2月, 2022 7 次提交
-
-
由 tanzhipeng 提交于
-
由 sneaxiy 提交于
-
由 Yan Chunwei 提交于
-
由 arlesniak 提交于
* amp list updated * tests updated * gray list updated * amp list updated * test updated
-
由 jakpiase 提交于
* Added adam kernel * CI rerun
-
由 Zhanlue Yang 提交于
-
由 Chen Weihang 提交于
* refactor custom op kernel func and utils * add output sync * adapte tensor* in utils * fix windows symbol error
-
- 06 2月, 2022 1 次提交
-
-
由 Wilber 提交于
-
- 04 2月, 2022 2 次提交
-
-
由 zyfncg 提交于
* add data_transform in pten api * support GetKernelTypeForVar * fix complie problem of bfloat16 * change error namespace * add complex type transform unittest * fix merge conflict
-
由 Chen Weihang 提交于
-
- 02 2月, 2022 3 次提交
-
-
由 Zuza 提交于
-
由 Chen Weihang 提交于
* remove kernel alias name * fix depreacted error * fix deprecated failed * fix mean error * resolve conflict * fix windows failed
-
由 Jiabin Yang 提交于
-
- 30 1月, 2022 10 次提交
-
-
由 Xiaoxu Chen 提交于
* add multinomial probability distribution * fix categorical sample bug when logits less than zero * fix categorical sample can't pass hypothesis test and entropy shape error bug
-
由 zhaocaibei123 提交于
* geo depends * add memory geo table * fix
-
由 zhangkaihuo 提交于
* dense_to_sparse_coo * optimize unit testing; support rocm * 1. delete fluid related header file 2. update the copyright * fix hipMemcpy * update dense_to_sparsecoo * add namespace sparse * sparse_csr_to_dense * test to_sparse_coo: csr_to_coo * fix writing error
-
由 Chen Weihang 提交于
* change unary infermeta * change other infermeta * change all infermeta format * resolve conflit * fix test failed * resolve reshape conflit * fix compile failed * adapt auto api gen * fix reshape failed * fix concat failed * resolve conflict
-
由 Chen Weihang 提交于
-
由 zhangkaihuo 提交于
* dense_to_sparse_coo * optimize unit testing; support rocm * 1. delete fluid related header file 2. update the copyright * fix hipMemcpy * update dense_to_sparsecoo * add namespace sparse
-
由 Leo Chen 提交于
-
由 fwenguang 提交于
-
由 mhhhh1 提交于
-
由 Leo Chen 提交于
* upgrade _get_all_register_op_kernels * add ut * support xpu/npu * fix device id * enhance TransToFluidPlace * fix compile
-