未验证 提交 a8981be0 编写于 作者: L Leo Chen 提交者: GitHub

[NewIR] support c_broadcast (#56284)

* [NewIR] support c_broadcast

* add legacyOpList
上级 12547fb4
...@@ -22,7 +22,7 @@ const std::unordered_set<std::string> LegacyOpList = { ...@@ -22,7 +22,7 @@ const std::unordered_set<std::string> LegacyOpList = {
"pd.fused_softmax_mask_upper_triangle_grad", "pd.fused_softmax_mask_upper_triangle_grad",
"pd.load_combine", "pd.load_combine",
"pd.c_concat", "pd.c_concat",
"pd.load_combine"}; "pd.c_broadcast_"};
enum class AttrType { enum class AttrType {
UNDEFINED = 0, UNDEFINED = 0,
......
...@@ -123,6 +123,16 @@ ...@@ -123,6 +123,16 @@
backward : batch_norm_grad backward : batch_norm_grad
optional : reserve_space optional : reserve_space
- op : c_broadcast
args : (Tensor x, int ring_id=0, int root=0, bool use_calc_stream=false)
output : Tensor(out)
infer_meta :
func : UnchangedInferMeta
param : [x]
kernel :
func : c_broadcast
inplace : (x -> out)
- op : c_concat - op : c_concat
args : (Tensor x, int rank, int nranks, int ring_id, bool use_calc_stream, bool use_model_parallel) args : (Tensor x, int rank, int nranks, int ring_id, bool use_calc_stream, bool use_model_parallel)
output : Tensor(out) output : Tensor(out)
......
...@@ -2990,6 +2990,12 @@ ...@@ -2990,6 +2990,12 @@
yolo_loss : GetYoloLossExpectedKernelType yolo_loss : GetYoloLossExpectedKernelType
yolo_loss_grad : GetYoloLossExpectedKernelType yolo_loss_grad : GetYoloLossExpectedKernelType
- op: c_broadcast
inputs :
x : X
outputs :
out : Out
- op: channel_shuffle - op: channel_shuffle
inputs: inputs:
{x: X} {x: X}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册