- 10 2月, 2022 9 次提交
-
-
由 chenyanlann 提交于
-
由 hong 提交于
* move masked select cpu kernel * add masked selected gpu kernel; test=develop * fix bugs; test=develop * bug fix; test=develop * bug fix; test=develop * add namespace to set mask array; test=develop * fix bug; test=develop * fix bugs; test=develop * fix ddim bug; test=develop * fix npu op bug; test=develop * fix xpu dependecy bug; test=develop * move kernel args to sig.cc; test=develop
-
由 wenbin 提交于
* mkldnn conv fix * definetion
-
由 Zhanlue Yang 提交于
-
由 crystal 提交于
* optimize conv1d forward * add conv opt * Optimize memory copy * delete share data with * set num_filters=512 * add nlc optimize * Optimize num_filter=512 data on A100 and V100 * Fix the workspace_size size setting of filter
-
由 zhangbo9674 提交于
* add squeeze unsqueeze stack * add unittest * add cpu kernel
-
由 zhangbo9674 提交于
* add dropout * add reshape * add slice * refien slice unittest * refine slice unittest * add cpu bf16 kernel
-
由 Leo Chen 提交于
* update isnan registration * fix compile
-
由 Aganlengzi 提交于
-
- 09 2月, 2022 19 次提交
-
-
由 Zhang Zheng 提交于
* Optimize performence of softmax_fwd when axis!=-1 * use functor * support hip * fix functor
-
由 Leo Chen 提交于
* fit pten for amp * fix typo
-
由 Wangzheee 提交于
* rebuild matmul pass: trt and gpu_cpu * rebuild matmul pass: trt and gpu_cpu * rebuild matmul pass: trt and gpu_cpu * rebuild matmul pass: trt and gpu_cpu
-
由 niuliling123 提交于
-
由 mhhhh1 提交于
-
由 fwenguang 提交于
-
由 fwenguang 提交于
-
由 fwenguang 提交于
-
由 Jiabin Yang 提交于
* merge legacy to fluid * Remove legacy code * Remove legacy code * Remove DataType test * Using Tensor directly instead of using EagerTensor * support gradient_accumulation * make test_imperative_lod_tensor_to_selected_rows longer * make test_imperative_lod_tensor_to_selected_rows longer
-
由 Yiqun Liu 提交于
-
由 hong 提交于
* add trace op * bug fix * bug fix; test=develop * thrust bug fix; test=develop * remove useless register; test=develop * fix bug; test=develop * update trace kernel; test=develop * move kernel args to trace_sig; test=develop
-
由 Chen Weihang 提交于
* fix slice bug of cusstom op * add offset in check
-
由 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 提交于
-
由 huzhiqiang 提交于
-
由 qipengh 提交于
-
由 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 12 次提交
-
-
由 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
-
由 yaoxuefeng 提交于
add downpour_ctr_accessor
-
由 yaoxuefeng 提交于
-
由 niuliling123 提交于
* Replace clip, bce_loss, full and full_like with elementwise
-