- 27 8月, 2021 2 次提交
- 26 8月, 2021 9 次提交
-
-
由 Jacek Czaja 提交于
[oneDNN] disable caching oneDNN primitives in matmul v2, Reduce grad and elementwise_add grad, expand_v2 (#35132) * - grad caching disabled of matmul_v1 - compilation fix - compilation fix * - reduction removed * - Matmul v2 disabled caching * Draft of further changes * - workaround for reducegrad * - fixes to UT * - fix to compilation * - another fix * - fix
-
由 duanboqiang 提交于
* fix assign bug support fp16 uint8 * fix dygragh assign bool bug * modify code style * revoke bool modification
-
由 smallv0221 提交于
* Support dropout backward in eval mode * add downscale case * minor fix * minor fix
-
由 Aurelius84 提交于
* Modify into QueueSync QueueAsync * fix complie on MacOS * fix pointer * fix conflict * polish unittest * fix windows fetch error * polish code according reviewer * fix device_guard on CPU place
-
由 Li Min 提交于
Describe Add feed_forward for fused attention op. (1) Encapsulate matmul impl (forward and backward) used in attention op. (2) Implement bias_add (forward and backward) used in attention op.
-
由 Bo Liu 提交于
-
由 shiyutang 提交于
* add_roi_align_npu * update * update * update
-
由 duanboqiang 提交于
-
由 XGZhang 提交于
-
- 25 8月, 2021 6 次提交
-
-
由 jakpiase 提交于
* temporary change * fix for expand_v2 * changes after review, activated ppyolov inference test
-
由 zhaoyingli 提交于
-
由 ronnywang 提交于
-
由 Yuang Liu 提交于
-
由 ronnywang 提交于
-
由 taixiurong 提交于
-
- 24 8月, 2021 8 次提交
-
-
由 gongweibao 提交于
-
由 wanghuancoder 提交于
* add fetch, test=develop * fix fetch2op, test=develop * fix fetch2op, test=develop * refine, test=develop * fix fetch ctx, test=develop * add wait, test=develop * rename fetch2 to fetch_v2, test=develop * merge, test=develop
-
由 duanboqiang 提交于
* fix bmm bug * bmm style * fix bmm
-
由 Jacek Czaja 提交于
* - concat refactoring draft * - cmpilation fixes * - yet another compilation fix * - fix * - compilation fix * - fixes to compilation * - another compilation fix * - fix * - Added overloaded AcquirePrimitiveDesc for concat * - fix * - reserve introduced * - UT fixes * - test concat int8 improved * - fixes * - fix to crash * - lint fixes * - fixes after review * - some other fixes from review
-
由 王明冬 提交于
-
由 ronnywang 提交于
* add conv_op_npu and test * add more tests * clean headers & support fp16 * update
-
由 ronnywang 提交于
* add pool2d_op_npu and test * update * update pool2d_backward_navie * clean headers
-
由 TeslaZhao 提交于
-
- 23 8月, 2021 7 次提交
-
-
由 Jacek Czaja 提交于
* - disabled interpolate onednn * - compilation fix * - draft of batch_norm cache disabling * - fixes to UT
-
由 Li Min 提交于
Refactor the organization of layer_norm cuda impl so that it can be reused in fused attention op. Extract the layer_norm cuda impl form layer_norm_op.cu to layer_norm_kernel.cu.h. Define fused/attention_layer_norm.h, which can be used in fused attention op in next PR.
-
由 zyfncg 提交于
* Support getitem by Bool index * delete some debug info of bool index * support the case that the shape of bool index is different from indexed tensor
-
由 pangyoki 提交于
-
由 pangyoki 提交于
-
由 TeslaZhao 提交于
-
由 zhaoyingli 提交于
* adamw support cuda * adamw support cuda
-
- 22 8月, 2021 1 次提交
-
-
由 Zhang Zheng 提交于
-
- 20 8月, 2021 7 次提交
-
-
由 Hao Lin 提交于
-
由 Yuang Liu 提交于
-
由 lzzyzlbb 提交于
* add rmsprop npu * add argsort npu * add argsort npu * modify according to review * modify sharedatawith according to review * modify reshape according to review * rm dygraph=false
-
由 Sing_chan 提交于
* [NPU] Support npu kernel for pad3d op * fix for comment of zhouwei25 * fix some bugs according to qili93's comments * add support and test for paddings in input * delete VLOG used for debug
-
由 zhaoyingli 提交于
* add depthwise_conv2d npu * add some tests * Delete test_unique_op_npu.py * delete trans input
-
由 zhaoyingli 提交于
* [NPU] Support npu op where and where grad * fix use const_cast * delete a test
-
由 JYChen 提交于
* add (N,C,*) input support for GroupNorm * --amend
-