- 05 6月, 2023 9 次提交
-
-
由 PommesPeter 提交于
* feat: added polygamma init code * feat: added polygamma unittest code * test: added more test cases * refactor: added forward impl * refactor: added backward impl * test: updated cases * refactor: updated test cases * refactor: added more case and fixed some bugs * test: updated ref func * refactor: updated code style * refactor: move the code * refactor: updated test * refactor: updated test * docs: updated en doc Co-authored-by: Nzachary sun <70642955+sunzhongkai588@users.noreply.github.com> * docs: updated math eq --------- Co-authored-by: Nzachary sun <70642955+sunzhongkai588@users.noreply.github.com>
-
由 gouzil 提交于
-
由 wangzhen38 提交于
-
由 houj04 提交于
-
由 umiswing 提交于
-
由 huangjiyi 提交于
Support code generation for op conv2d_transpose, conv3d_transpose, depthwise_conv2d_transpose (#54242)
-
由 huangjiyi 提交于
* update * update eager_gen * update * rm intermediate
-
由 Asthestarsfalll 提交于
* optimize logsumexp in small data scale * fix * fix * add #pragma once * swith to use aligned_vector and support arbitrarily shape * fix store * fix store * refine for special cases * try * fix * update * fix * fix all_reduce * try * fix rocm bug * fix rocm bug * fix rocm bug * fix rocm bug * fix rocm bug * fix rocm bug * fix rocm bug * fix rocm bug
-
由 Chen Weihang 提交于
-
- 03 6月, 2023 2 次提交
-
-
由 RedContritio 提交于
* remove reduce_sum_op.h * support auto generate for static op reduce_sum * remove reduce_sum_op in CMakeLists.txt
-
由 Scotty 提交于
-
- 02 6月, 2023 14 次提交
-
-
由 kangguangli 提交于
* add vector type support for program translator * polish * support basic attribute type * resolve conflicts * add verify for combine/slice and unittests * polish * support more type in attribute translator * modify by reviews * fix merge mistakes * refine code * refine code * add interface * fix: op name normalization * fix typo * refactor input translator * fix merge conflicts * fix op normalizer bug * refactor attribute translator * fix bug * refactor output translator * fix typo * fix * fix approval error * fix coverage * fix op_compat parser * fix merge conflicts * fix merge conflicts * fix merge conflicts * fix merge conflicts * fix merge conflicts * revert some changes --------- Co-authored-by: Nzhangbo9674 <zhangbo54@baidu.com>
-
由 RedContritio 提交于
-
由 RedContritio 提交于
* decouple reduce_any_op.h and reduce_op.h from reduce_any_op.cc * support auto generate for static op reduce_any
-
由 RedContritio 提交于
-
由 RedContritio 提交于
-
由 YuanRisheng 提交于
* fix mac typeinfo bugs * add file * move code to cc * fix compile bugs
-
由 Difer 提交于
* add fp&bf16 bernoulli * add check_dtype & fix error * fix rocm error
-
由 wz1qqx 提交于
-
由 Hui Zhang 提交于
* floor div support float/double/bfloat16/float16 * add ut * fix bug * fix fft.ifftshift for floor_divide upgrade * fix comment * fix bugs * fix bug
-
由 Zhang Zheng 提交于
* Optimize perf of broadcast matmul * support more dtype
-
由 傅剑寒 提交于
-
由 Zhang Ting 提交于
* support master_grad for adam and momentum Co-authored-by: zhangting_2017@163.com <zhangting2020>
-
由 Wang Xin 提交于
* static graph autogen code for shape op * fix onednn * fix onednn
-
由 xiaoguoguo626807 提交于
-
- 01 6月, 2023 8 次提交
-
-
由 zhangbo9674 提交于
* add build * add build * refine code * refine code * refine code * refine code * refine interface * fix bug * fix bug * fix bug * refine yaml
-
由 umiswing 提交于
-
由 zhouweiwei2014 提交于
-
由 ronnywang 提交于
* [ROCM] fix multihead_matmul * skip bf16 uts * update
-
由 Wang Xin 提交于
* static graph autogen code for check_finite_and_unscale_ op * bug fixed
-
由 YuanRisheng 提交于
-
由 RedContritio 提交于
-
由 huangjiyi 提交于
* update * update cmake * update * update * update * update * Revert "update cmake" This reverts commit 1e1dc1b2bc9967b725201272607f939260070fd4. * update * update * update * update
-
- 31 5月, 2023 3 次提交
-
-
由 RedContritio 提交于
* support auto generate for static op reduce_amax * set reduce_amax attr 'axis' type as IntArray
-
由 YuanRisheng 提交于
* fix inference static lib bugs * add if for copy * fix py3 bugs
-
由 Charles-hit 提交于
* support activation prim op bf16 dtype * remove useless code
-
- 30 5月, 2023 4 次提交
-
-
由 risemeup1 提交于
* update_c++17 * update_c++17 * fix windows bug * solve cirle depend * solve cirle depend * solve cirle depend * solve cirle depend * solve cirle depend * fix windows bug * fix compiler error * fix compiler error * update eigen3 * update eigen3 * update eigen3 * fix mac-py3 compiler error * update C++17 * fix mac compiler error * fix compile error * fix coverage_compiler error * fix coverage_ci_problem * fix coverage_error * fix_kunlun200 compile error * fix kunlun200 compiler error * fix compile error * fix compiler error * fix py3 failed test * fix kunlun200 compiler error * test * fix test error * fix test error * fix test error * test * test * fix mac py3 error * fix mac py3 error * fix mac py3 error * fix test error * fix test error * fix compile error * fix compile error * fix compile error * test * test * fix compiler error * test * test * debug on ci * fix compiler error * fix compiler error * test * fix cinn compiler error * test * fix rocm cmpile error * fix cinn and kunlun compile error * update c++14 * Update flags.cmake
-
由 shaojie_wang 提交于
* softmax fwd: force vec size to 1 when dtype is float * use 1024 as threshold to use cudnn
-
由 Yiqun Liu 提交于
* Reimplement the check_nan_inf function as check_numerics kernel. * Remove the cpu implemention to phi. * Add ifdef for the including of omp.h. * Move the use of FLAGS_check_nan_inf_level out of header file. * Implement a common PrintAndThrowError function. * Fix the error using of __NVCC__, which should be instead with __CUDA_ARCH__. * Add dependency of phi. * Polish codes and unittest.
-
由 Leo Chen 提交于
* add timer to log deps * rename flag * add ut
-