- 13 1月, 2023 30 次提交
-
-
由 limingshu 提交于
* first commit * add some changes in stack kernel. * move the location of GeneralDivMod * fix code format error according to ci
-
由 cyber-pioneer 提交于
-
由 Wang Bojun 提交于
* add fmha_flashattention oss plugin
-
由 zhouweiwei2014 提交于
-
由 wanghuancoder 提交于
-
由 ronnywang 提交于
* add where, atan2, median 0d ut * add where, atan2, median 0d ut * update * update * update
-
由 Zhang Jun 提交于
* update trt engine to set in/out data type * update * Update engine.cc * Update engine.cc * update * set engine output type before freeze the network * update * update trt autoscan ut * update * update ut * fix equal bug, update ut * fix cast and equal ut * update cast ut using TRT < 8.4 * set datatype from scope * check output var is nullptr * Update op_converter.h * update tensorrt_engine_op_test ut * update
-
由 duanyanhui 提交于
* clear ProcessGroupCustom manually * fix bug * fix bug * move destroy ProcessGroup to ProcessGroupIdMap * enable destroy to all device * remove unused comments * change to internal api * Update process_group.cc * Update process_group.cc
-
由 duanyanhui 提交于
* update get_device to custom * add custom_device api * rm is_compiled_with_custom_device from framework * add todo comments
-
由 Jiabin Yang 提交于
* support elementwise base func * fix compiling error and add test * remove additional param * support vjp for div using comp * remove additional change * fix dy2st error with magic num * fix dy magic num * another magic * another magic * add more test * fix windows problem * another magic * fix windows compile * invoke ci * add skip rename strategy * support add vjp * fix test_tanh * support add with new axis cal * fix resnet and some test * add composite log * support sub vjp
-
由 wuhuachaocoding 提交于
-
由 Weilong Wu 提交于
* [Eager] polish some apis logic * polish api logic
-
由 wuhuachaocoding 提交于
-
由 risemeup1 提交于
-
由 LiYuRio 提交于
-
由 jameszhang 提交于
* kunlun add support for c_concat and c_split * replace mutable_data() and ShareDataWith()
-
由 ykkk2333 提交于
-
由 jameszhang 提交于
* fix xpu unittest issue: zero_dim_tensor * deal with leftout issue introduced by #49470
-
由 Leo Guo 提交于
-
由 zhouweiwei2014 提交于
-
由 wangshengxiang 提交于
-
由 zyfncg 提交于
* generate static graph code of stack, unbind, unique_consecutive op * fix bug
-
由 Wilber 提交于
-
由 wangzhen38 提交于
* [cpplint fix] under ps
-
由 Weilong Wu 提交于
* [PHI] rrelu add yaml * polish * polish
-
由 wuhuachaocoding 提交于
-
由 HongyuJia 提交于
-
由 zhangkaihuo 提交于
-
由 Weilong Wu 提交于
* [Eager] polish some api logic * fix split * revover
-
由 Yuanle Liu 提交于
-
- 12 1月, 2023 10 次提交
-
-
由 xiaoxiaohehe001 提交于
-
由 sunli 提交于
* lerp support 0 Tensor * fix lerp grad * fix lerp zero test * fix 0D + ND/ND + 0D * fix check * update code * fix lerp infer shape * static backward test * updata static graph test
-
由 Wen Sun 提交于
* refactor: migrate comm checks * refactor: add check in comm context * feat: add gloo static check * refactor: add place param in static check
-
由 zhangkaihuo 提交于
-
由 jameszhang 提交于
* Fix reduce func bug in process_group_bkcl Also catch up with a recent process_group PR that failed to add XPU branch. Note that reduce is still accomplished by allreduce for xpu. Fix this should xccl lib be updated. * fix compile issue for non-XPU
-
由 YuanRisheng 提交于
-
由 tianshuo78520a 提交于
* test=ninja;test=document_fix * test=ninja;test=document_fix * test=ninja;test=document_fix * add ninja * update dockerfile * update dockerfile * update dockerfile * update dockerfile * update dockerfile * test=cuda117 * update ce dockerfile * update ce dockerfile
-
由 gem5 提交于
-
由 xiaoxiaohehe001 提交于
-
由 wenbin 提交于
* compile fix * fix compile * compile fix * add more preln
-