mindspore2022/mindspore/ccsrc/kernel
caifubi 7d07e17f5a Support Common Hccl Op
1.Support Broadcast op
2.Support communication op as graph output
3.Optimize Communication op memory alocation
4.support hccl multi-group
2020-05-18 14:48:34 +08:00
..
aicpu save graph and task desc info to framework file 2020-05-11 09:41:40 +08:00
akg remove ge depend in cpu 2020-04-02 19:38:42 +08:00
cpu cpu kernel register type 2020-05-09 17:31:45 +08:00
gpu LayerNormGrad fix & codex 2020-05-15 11:16:30 +08:00
hccl Support Common Hccl Op 2020-05-18 14:48:34 +08:00
kash initial version 2020-03-27 22:54:54 +08:00
oplib add attr op_pattern to kernel build info 2020-05-14 14:41:27 +08:00
rts !1113 rename kernel/mng to kernel/rts 2020-05-15 09:39:18 +08:00
tbe !1129 Add op bounding box encoder 2020-05-18 09:04:59 +08:00
CMakeLists.txt !1113 rename kernel/mng to kernel/rts 2020-05-15 09:39:18 +08:00
ascend_kernel_mod.h save graph and task desc info to framework file 2020-05-11 09:41:40 +08:00
common_utils.cc fix pylint and codedex warnings 2020-04-29 18:35:20 +08:00
common_utils.h change directory of akg cuda kernel 2020-04-28 18:53:18 +08:00
kernel.h add attr op_pattern to kernel build info 2020-05-14 14:41:27 +08:00
kernel_build_info.cc add attr op_pattern to kernel build info 2020-05-14 14:41:27 +08:00
kernel_build_info.h add attr op_pattern to kernel build info 2020-05-14 14:41:27 +08:00
kernel_fusion.cc support buffer fusion 2020-04-28 20:25:15 +08:00
kernel_fusion.h initial version 2020-03-27 22:54:54 +08:00
kernel_query.cc !1162 change using input or output's device dtype to create transdata's kernel build info 2020-05-15 18:09:17 +08:00
kernel_query.h using device dtype to create transdata kernel build info 2020-05-15 16:10:17 +08:00
task_stream.h initial version 2020-03-27 22:54:54 +08:00