- 09 2月, 2022 10 次提交
-
-
由 Chen Weihang 提交于
-
由 hong 提交于
* update basic infrastructure; support string, suport vecotr<int>, add tensor args type index; test=develop * remove useless code; test=develop * fix bug; test=develop * polish code; test=develop
-
由 sneaxiy 提交于
-
由 Zhang Zheng 提交于
-
由 niuliling123 提交于
-
由 huzhiqiang 提交于
-
由 qipengh 提交于
-
由 zhangkaihuo 提交于
* implement AllocateFrom * 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 * to_sparse_csr: dense_to_sparse_csr and sparse_coo_to_csr * fix check shape * fix unit test * replace CUDADeviceContext by GPUContext
-
由 Tomasz Socha 提交于
-
由 hong 提交于
* add norm cpu * update code; * norm bug fix * move norm op to pten; test=develop * move norm op to pten; test=develop * add norm util; test=develop * fix norm npu bug; test=develop * fix norm kernel bug; test=develop * move kernel args to pten; test=develop * move kernel args to pten sig; test=develop
-
- 08 2月, 2022 23 次提交
-
-
由 Yan Chunwei 提交于
-
由 sneaxiy 提交于
* add more int id type support for embedding * add ut * add more ut * fix ci error
-
由 hlygit66666 提交于
* add fuse_relu_depthwise_conv_pass unittest * fix atol and rtol * fix according to review * Add FuseOptimizerPass and fuse_adam_pass unittest * add sgd and momentum unittest * add fuse_optimizer_pass * close amp * close amp * update * fix run on two cards * Update test_dist_fuse_adam_pass.py * Update test_dist_fuse_momentum_pass.py * Update test_dist_fuse_sgd_pass.py * Create test_dist_fuse_sgd_pass.py * Create test_dist_fuse_sgd_pass.py * Create test_dist_fuse_sgd_pass.py * Update test_dist_fuse_adam_pass.py * Update test_dist_fuse_momentum_pass.py * Update test_dist_fuse_sgd_pass.py
-
由 Yiqun Liu 提交于
-
由 joanna.wozna.intel 提交于
* Fix quantization next op findings * Corrections according to the review
-
由 Jacek Czaja 提交于
* - 38126 potential fix * - fix * - build fix * - another candidate fix * - compilation fix * - another fix * - Fix to activation of NHWC being first oneDNN op in chain on oneDNN ops * - compilation fix * - added NHWC reotating for elementwise being first op * - compilation fix * - compilation fix * - Added UT * - cosmetic fixes
-
由 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
-