hanhuifeng2020
bc46d644fe
[GraphKernel]Support reshape/elewise/broadcast+transdata fusion
2021-05-08 11:42:14 +08:00
mindspore-ci-bot
0887d35b1c
!15951 [GraphKernel]add the attribute reduce_output_fuse to enable fuse for the reduce_output on Ascend
...
From: @hanhuifeng2020
Reviewed-by: @gaoxiong1,@dylangeng
Signed-off-by: @dylangeng
2021-05-07 11:38:06 +08:00
mindspore-ci-bot
ed539597c2
!15415 [GraphKernel]adapt for logsoftmax in ascend
...
From: @wenfangpei
Reviewed-by: @gaoxiong1,@ckey_dou,@gaoxiong1,@ckey_dou
Signed-off-by: @ckey_dou,@ckey_dou
2021-05-07 11:19:58 +08:00
wenfangpei
4174a7b38f
expanders of some fusion ops
2021-05-06 10:23:41 +08:00
wenfangpei
db8256e61f
adapt for logsoftmax in ascend
2021-05-06 10:13:28 +08:00
mindspore-ci-bot
1827697642
!15961 Eliminate recursion call in split model
...
From: @gaoxiong1
Reviewed-by: @ckey_dou,@dylangeng
Signed-off-by: @dylangeng
2021-05-06 10:06:35 +08:00
wenfangpei
c41875b318
adapt expanders of some ops from gpu to ascend
2021-05-04 16:32:45 +08:00
Gaoxiong
4bc67f38de
eliminate recursion call
2021-04-30 16:26:27 +08:00
hanhuifeng2020
425d401e85
[GraphKernel]add the attr reduce_output_fuse to enable fuse for reduce_output on Ascend
2021-04-30 11:48:23 +08:00
mindspore-ci-bot
52e7f51970
!15741 [GraphKernel] batchnorm expander supports when first input is float16
...
From: @looop5
Reviewed-by: @gaoxiong1,@dylangeng
Signed-off-by: @dylangeng
2021-04-27 16:27:29 +08:00
mindspore-ci-bot
ff75ce8ac4
!15097 [graph kernel] clean code for expanders.
...
From: @chenlei_autodiff
Reviewed-by: @gaoxiong1,@dylangeng
Signed-off-by: @dylangeng
2021-04-27 16:26:18 +08:00
looop5
24f441ba33
batchnorm expander supports when first input is float16
2021-04-27 09:46:42 +08:00
mindspore-ci-bot
d6f58cb765
!15658 Reduce recursion overhead of split model
...
From: @gaoxiong1
Reviewed-by: @anyrenwei,@dylangeng
Signed-off-by: @dylangeng
2021-04-26 16:39:04 +08:00
chenlei_autodiff
fd227bb448
[graph kernel] clean code for expanders.
2021-04-26 16:28:37 +08:00
mindspore-ci-bot
168c64b60d
!15648 [GraphKernel] negative axis in Squeeze expander.
...
From: @chenlei_autodiff
Reviewed-by: @gaoxiong1,@anyrenwei
Signed-off-by: @anyrenwei
2021-04-26 14:18:21 +08:00
Gaoxiong
71002ed19d
reduce recursion overhead of split model
2021-04-25 21:27:09 +08:00
zengzitao
8dcff8d83c
refactor tile op and in expander open on gpu
2021-04-25 17:42:37 +08:00
chenlei_autodiff
b419f60b0d
[GraphKernel] negative axis in Squeeze expander.
2021-04-25 17:31:04 +08:00
r1chardf1d0
5c5d125b1d
optimize stitch fusion strategy
2021-04-25 10:58:19 +08:00
hanhuifeng2020
25505642ce
enable GraphKernel for TransData
2021-04-16 16:23:15 +08:00
lingyunli63
c48c2430f0
fuse matmul and elementwise in graphkernel
2021-04-14 20:52:35 +08:00
mindspore-ci-bot
cd002cb7f7
!14893 enable stitch fusion on bert
...
From: @r1chardf1d0
Reviewed-by: @gaoxiong1,@ckey_dou
Signed-off-by: @ckey_dou
2021-04-12 10:39:00 +08:00
mindspore-ci-bot
5b4685c5ea
!14604 [GraphKernel] add some expander ops
...
From: @chenlei_autodiff
Reviewed-by:
Signed-off-by:
2021-04-12 10:33:53 +08:00
r1chardf1d0
3b32995936
enable stitch fusion on bert
2021-04-09 18:29:55 +08:00
chenlei_autodiff
13fbfca6b9
[graph kernel] add expander ops.
2021-04-09 17:33:06 +08:00
wenfangpei
b9715db358
bugfix in expanders of layernorm
2021-04-09 17:31:32 +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
wenfangpei
83399c1b8d
adapt for layermorm C++ code
2021-04-06 11:06:50 +08:00
wenfangpei
a4ad6066b1
expander lamb_apply_weight_assign
2021-04-02 17:28:59 +08:00
mindspore-ci-bot
ddf75da542
!14085 [GraphKernel] add some expander ops
...
From: @chenlei_autodiff
Reviewed-by:
Signed-off-by:
2021-03-30 15:33:06 +08:00
mindspore-ci-bot
75fdaaa6aa
!14304 [GraphKernel] Dump GraphKernel split info as text; dump akg kernel launch fail message
...
From: @dayschan
Reviewed-by: @gaoxiong1,@gaoxiong1,@anyrenwei
Signed-off-by: @anyrenwei
2021-03-30 15:32:14 +08:00
chenlei_autodiff
f4289d40f3
add graph kernel expander ops.
2021-03-30 11:19:40 +08:00
dayschan
3c6c30024c
dump graph_kernel_split info
2021-03-29 17:15:02 +08:00
tronzhang
87bf1ec80f
delete mark_interface_fusion and tensor reuse frontend pass for graph kernel
2021-03-27 11:45:04 +08:00
lingyunli63
4b966ed40d
support matmul on D
2021-03-26 15:57:26 +08:00
huangbingjian
72ae1799f3
remove control_depend from py file
2021-03-22 20:45:45 +08:00
mindspore-ci-bot
5b95409022
!13512 add some expander ops
...
From: @zengzitao
Reviewed-by:
Signed-off-by:
2021-03-19 15:00:23 +08:00
mindspore-ci-bot
2fadad0875
!13121 expander lamb_apply_optimizer_assign
...
From: @wenfangpei
Reviewed-by:
Signed-off-by:
2021-03-19 14:59:37 +08:00
wenfangpei
043a558ae2
expander lamb_apply_optimizer_assign
2021-03-19 09:54:10 +08:00
zengzitao
d0a656f3cd
add some expander ops
2021-03-18 19:03:36 +08:00
dayschan
a2967330ea
Normalize the Reduce nodes' axis in GraphKernel
...
changed all reduce axis to non-negative value
2021-03-16 15:55:50 +08:00
tronzhang
7252ffb66b
pipeline optimization for parallel fusion
2021-03-09 17:07:24 +08:00
mindspore-ci-bot
d285692217
!12852 Change maketuple in coreops
...
From: @liangzhibo
Reviewed-by:
Signed-off-by:
2021-03-05 19:13:42 +08:00
l00591931
680324f225
Change make tuple in core.ops
2021-03-05 10:54:33 +08:00
dayschan
454500309c
add OpInfer for op Select
2021-03-05 10:21:28 +08:00
dayschan
7beca18f3c
Refactor GraphKernelExpander (3rd submission)
...
Rewrited the OpInfer as a class with functions "infer_shape","infer_type"
and "infer_format". Used op name to find the subclass that have implemented
these functions. Two common class "_Elemwise" and "_Reduce" was provided.
Op BiasAddGrad supported "FractalNZ" format.
2021-02-25 19:14:46 +08:00
dayschan
9d572f3963
Refactor GraphKernelExpander (2nd submission)
...
Changed the expander to class, and used the class name to identify the operator.
Moved the original op logic into the `_expand` function; added a `_check` function to check op inputs.
Use decorator to register the whitelist formats that operator supports,
and the decorator will change the `_check` function to check the formats.
2021-02-22 14:49:21 +08:00
dayschan
e0e6c39eae
Refactor GraphKernelExpander (1st submission)
...
Decoupled from the front-end interfaces.
1. Removed the call to "Renormalize".
Completed the infer-format in model_builder.
Only used the device shape and device format to
infer an abstract shape without considering padding.
2. Removed the call to python's Primitive interfaces.
The "Renormalize" relies on the PrimitivePy, so they can be
removed together. After that, the functions "ConstAttrToInput",
"DeleteAttrInInput" and related can be removed.
3. Reuse the AkgKernelJsonGenerator in GraphKernelExpander.
1) set the attribute "extract_opinfo_from_anf" to true, so that
the generator can handle the basic operator with anf info.
2) added a function "extract_expand_info" in expander.py
to convert the json into a more friendly format. The attrs
was converted to a dict instead of a list.
4. Scalars only support DefaultFormat.
Removed the argument "format" from graph_builder.value
5. Moved the expander op list from graph_kernel_helper.cc to graph_kernel_expander.cc
2021-02-07 10:48:23 +08:00
jinyaohui
30a27b2adb
modify Gelu、FastGelu to GeLU and FastGeLU
2021-02-05 17:19:52 +08:00