looop5
ae5532fb9e
pass lambda function to out_shape and out_dtype
2021-12-06 10:21:16 +08:00
Yang Jiao
5acce31c72
apply autodiff in custom
2021-12-03 11:16:19 +08:00
looop5
9c4fe8b6bf
update akg, refine document of Custom op
2021-11-16 11:06:40 +08:00
looop5
58e27d87bc
add Custom, custom_op_info_register, CustomRegOp to __init__
2021-11-11 20:49:16 +08:00
Yang Jiao
2b112ef656
add custom pynative test cases
2021-11-08 17:37:39 +08:00
looop5
b89d744e80
Custom op supports no reg info
2021-11-05 17:21:39 +08:00
Yang Jiao
9bcf28d15d
adapt custom op to pyfunc kernel
2021-10-31 22:08:10 +08:00
looop5
0b34bf33bc
add custom op pynative testcases
2021-10-25 20:53:42 +08:00
Yang Jiao
40b648b873
add aot
2021-10-22 11:37:39 +08:00
looop5
46789f260a
Custom operator supports tbe dsl
2021-10-15 16:28:33 +08:00
huanghui
ba66c0d491
add security isolate for save_graphs
2021-09-14 10:24:12 +08:00
yanghaoran
a0ff742442
prune less effective Ascend testcases to level1
2021-08-27 19:59:59 +08:00
Zichun Ye
a7d89f6686
add graph kernel userdefine op support
...
fix code check
2021-08-25 14:37:28 +08:00
dayschan
9add26ad99
Add expanders in c++ code
...
transplant the op expander code from python to c++, base on LiteGraph.
the c++ expander will be called in priority if it was registered in OpExpanderFactory.
add two examples, BiasAdd and ExpandDims.
remove BiasAdd from python expanders.
since the ExpandDims is also imported by other ops (e.g. BatchNorm), we don't remove it now.
2021-08-12 16:22:56 +08:00
chenlei_autodiff
0271535429
[GraphKernel] fix bert and add graph kernel ops.
2021-08-10 16:02:00 +08:00
i-robot
9629b4ebd9
!21071 [Graph Kernel] Add Low Precision Optimition
...
Merge pull request !21071 from cyun/master_729
2021-08-05 00:50:58 +00:00
cy
4105a247b7
add low precison
2021-08-04 09:52:42 +08:00
Yang Jiao
fd7ab25fc2
reorganiz empty graph
2021-07-30 15:37:47 +08:00
Zichun Ye
22172f18bc
update graph kernel support for argmax/argmin
...
fix pylint problem
fix conflict
fix op list
fix check warning
fix code based on review comments
update akg commit
fix check warning
2021-07-30 09:53:12 +08:00
Yang Jiao
e2cfc516eb
refctor arithmetic simplify
2021-07-27 11:29:26 +08:00
lingyunli63
50a66ae476
isnan isfinite isinf squaresumall identity oneslik
2021-07-26 12:58:30 +08:00
i-robot
26027b89fe
!19806 add graph kernel erf, erfc, floor, floordiv, mod, floormod, div
...
Merge pull request !19806 from 杨林枫/basic_graph_kernels
2021-07-22 12:57:55 +00:00
yanglf1121
c30b1e6d06
add graph kernel div, floordiv, mod, floormod, floor
2021-07-22 14:16:56 +08:00
i-robot
7d2a07a2bd
!20480 Add LiteGraph for graphkernel
...
Merge pull request !20480 from DeshiChen/0618_litegraph
2021-07-22 01:31:45 +00:00
dayschan
137608b518
Add LiteGraph for graphkernel
...
Add a subdirectory "model" in the "backend/optimizer/graph_kernel" for litegraph.
Implement two interfaces "AnfGraph2LiteGraph" and "LiteGraph2AnfGraph".
The litegraph will be the base data structure when we migrate the GraphKernel code
from python("mindspore/_extends/graph_kernel") to c++.
2021-07-21 10:21:11 +08:00
wangrao124
7cddde47b0
!215 add ops: LogicalNot, LogicalAnd, LogicalOr, NotEqual, EqualCount, Asinh, Acosh
...
* add ops: LogicalNot, LogicalAnd, LogicalOr, NotEqual, EqualCount, Asinh, Acosh
2021-07-21 10:01:17 +08:00
chenlei_autodiff
7d55cef106
[GraphKernel] add sponge ops.
2021-07-19 19:34:23 +08:00
dayschan
149dab39c5
Add expander for AddN; update akg submodule
2021-06-29 11:43:46 +08:00
lingyunli63
a995bea507
recompute_fuse
2021-06-22 20:39:11 +08:00
zengzitao
8064de7931
fix maximum_grad and minimum_grad input_shape not equal to output_shape bug
2021-06-16 14:25:18 +08:00
i-robot
07f58b0b46
!17626 matmul to mul
...
Merge pull request !17626 from lingyunli63/matmul_to_mul
2021-06-10 11:35:18 +08:00
looop5
a5cb14261d
recover several graph kernel ci testcases
2021-06-09 20:32:02 +08:00
lingyunli63
4f34e537a0
replace matmul/batchmatmul by mul when k is 1
2021-06-09 15:00:27 +08:00
wenfangpei
fd84c20a6a
bug fix in lamb
2021-05-29 16:02:30 +08:00
mindspore-ci-bot
f91a365564
!16322 [GraphKernel] Enable matmul for gpu
...
From: @lingyunli63
Reviewed-by:
Signed-off-by:
2021-05-20 10:08:47 +08:00
lingyunli63
afc69b16f7
enable gpu gk MatMul and insert pad/unpad
2021-05-19 16:03:29 +08:00
r1chardf1d0
6609f706fe
update akg && set random seed for stitch case
2021-05-18 09:25:16 +08:00
r1chardf1d0
b1be842d2b
add stitch fusion case in ci
2021-05-08 15:17:14 +08:00
wenfangpei
db8256e61f
adapt for logsoftmax in ascend
2021-05-06 10:13:28 +08:00
mindspore-ci-bot
92355554ed
!15412 [GraphKernel]adapt expanders of some ops from gpu to ascend
...
From: @wenfangpei
Reviewed-by: @gaoxiong1,@ckey_dou
Signed-off-by: @ckey_dou
2021-05-06 09:42:15 +08:00
wenfangpei
c41875b318
adapt expanders of some ops from gpu to ascend
2021-05-04 16:32:45 +08:00
dingpeifei
d27ad7e797
upgrade_ascend_0428
2021-04-30 15:57:57 +08:00
looop5
e88cdc84ec
enhancement reorder_ops pass to support reordering cast and type insensitive operators
...
support castup, type-insensitive to type-insensitive, castup
refactor reorder_ops
fix compiling
move reorder_ops pass to later
fix abstract
refactor
fix node input num bug
2021-04-25 11:06:48 +08:00
lingyunli63
c48c2430f0
fuse matmul and elementwise in graphkernel
2021-04-14 20:52:35 +08:00
chenlei_autodiff
13fbfca6b9
[graph kernel] add expander ops.
2021-04-09 17:33:06 +08:00
wenfangpei
66d28af79e
adapt for layernorm in ascend
2021-04-08 18:11:59 +08:00
mindspore-ci-bot
b5bc938deb
!12914 [GraphKernel]expander lamb_apply_weight_assign
...
From: @wenfangpei
Reviewed-by: @anyrenwei,@gaoxiong1,@gaoxiong1
Signed-off-by: @anyrenwei
2021-04-06 17:02:16 +08:00
mindspore-ci-bot
f324a9a760
!14553 [GraphKernel] refine cast matmul fusion
...
From: @lingyunli63
Reviewed-by: @gaoxiong1,@ckey_dou
Signed-off-by: @ckey_dou
2021-04-06 11:07:55 +08:00
lingyunli63
56390330ac
cast_Matmul_fusion, when cast cannot fuse forward
2021-04-02 17:32:11 +08:00
wenfangpei
a4ad6066b1
expander lamb_apply_weight_assign
2021-04-02 17:28:59 +08:00