Commit Graph

15 Commits

Author SHA1 Message Date
huangbingjian 5a73a26fee remove control_depend from py file 2021-03-23 20:47:43 +08:00
l00591931 680324f225 Change make tuple in core.ops 2021-03-05 10:54:33 +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
mindspore-ci-bot a24ff36d9c !11777 stitch fusion
From: @r1chardf1d0
Reviewed-by: 
Signed-off-by:
2021-02-02 10:49:04 +08:00
l00591931 9ec100d069 Change TensorAdd to Add, from r1.1 to master 2021-02-01 17:53:52 +08:00
r1chardf1d0 9d6392c5c5 stitch info 2021-02-01 10:20:55 +08:00
looop5 848be9b07c add tile to expand list
add tile expander

add BroadcastTo in model

fix BroadcastTo op calling error and infer shape

rewrite tile expander

not split broadcast_to

add SqrtGrad expander
2020-12-05 10:57:55 +08:00
tronzhang 9d7494f4df split shape ops for more fusion pportunity. 2020-11-17 15:22:51 +08:00
dayschan 195b1fe8d5 Add Transpose into fusible list. 2020-11-11 11:02:06 +08:00
zengzitao 53043ae18f support expand fused_adam and fused_adam_weight_decay op 2020-11-05 20:56:49 +08:00
zengzitao 5cfa172720 expand gelu and gelugrad op 2020-11-02 11:35:01 +08:00
zengzitao febdb1850c expand bias_add and bias_add_grad op 2020-10-30 14:52:39 +08:00
chenzomi 44bf4c3e37 [ME] format code 2020-10-26 10:17:45 +08:00
lingyunli63 dd48f10c3d add assign ops in composite_topi 2020-09-28 10:20:53 +08:00
dayschan 37a48f6aac GraphKernel supports GPU
1. Update akg submodule
2. Refactor akg_kernel_build, akg_ascend_kernel_build, akg_gpu_kernel_build
3. Add akg_kernel_json_decoder to support converting kernel_json to AnfNode.
4. Add GraphKernel Cost Model. (mindspore/_extends/graph_kernel)
5. Add some GraphKernel passes to GpuSession, move these passes to backend/optimizer/graph_kernel.
6. Add global id for ir files.
7. Fix bug in ConstInputToAttr.
2020-09-09 17:46:50 +08:00