- 14 2月, 2022 1 次提交
-
-
由 chentianyu03 提交于
* add split kernel * add split kernel signature * fix split bug * modify MakePtenScalarArrayFromVarList * modify MakePtenScalarArrayFromVarList * fix split windows register error * add test case for split kernel * replace raw split kernel with pten kernel * fix makeScalar/ScalarArray bug * remove debug log * remove int64_t type in buildPtcontext * update by code review * fix split dev test failed * change DenseTensorMeta to MetaTensor * change split api code from auto gen to manual * split cuda kernel support bfloat16 type * fix conflict * rm raw split kernel * merge develop branch * change to pten::errors
-
- 11 2月, 2022 1 次提交
-
-
由 Chen Weihang 提交于
* move grad get expected pten kernel args * fix reduce sum error * fix element_sub_grad failed * revert kernel judge change
-
- 10 2月, 2022 2 次提交
-
-
由 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
-
由 Chen Weihang 提交于
* add standard_suffix_set_and_remove_reshape_with_xshape * revert reshape change * polish reduce name
-
- 09 2月, 2022 3 次提交
-
-
由 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
-
由 Leo Chen 提交于
-
由 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 1 次提交
-
-
由 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
-
- 02 2月, 2022 1 次提交
-
-
由 Chen Weihang 提交于
* remove kernel alias name * fix depreacted error * fix deprecated failed * fix mean error * resolve conflict * fix windows failed
-
- 28 1月, 2022 1 次提交
-
-
由 Chen Weihang 提交于
* update forward argument mapping * fix compile failed * fix test failed
-
- 26 1月, 2022 1 次提交
-
-
由 Chen Weihang 提交于
* infermeta context init design * support infermeta called in fluid op * add hasattr and attr methods * add dygraah GetVarPtrs support * rename arg_map_context to arg_map_utils * add registry for arg map func * resolve conflit * refactor op utils design * polish meta config * fix details * remove hasattr method * resolve conflit * revert cmake order change * revert some change * change init pos * fix compile faileed * fix typo * fix inference failed * fix windows ccompile failed * polish format Co-authored-by: NWang Huan <wanghuan29@baidu.com>
-
- 22 1月, 2022 1 次提交
-
-
由 Chen Weihang 提交于
* add attr for arg map context * add argument fn declare * add attr test for get attr value method * polish details
-
- 10 1月, 2022 1 次提交
-
-
由 Chen Weihang 提交于
-
- 09 12月, 2021 1 次提交
-
-
由 Chen Weihang 提交于
-