Skip to content

GitLab

Commit 60f852b3 authored by Paul's avatar Paul
Browse files

Format

parent d976ecfe
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
src/simplify_algebra.cpp
View file @ 60f852b3
...@@ -298,8 +298,8 @@ struct find_dot_mul ...@@ -298,8 +298,8 @@ struct find_dot_mul
return; return;
} }
auto broadcast_v = d_ins->get_operator().to_value(); auto broadcast_v = d_ins->get_operator().to_value();
auto c_lens = c_ins->get_shape().lens(); auto c_lens = c_ins->get_shape().lens();
std::vector<int64_t> permutation(c_lens.size()); std::vector<int64_t> permutation(c_lens.size());
std::iota(permutation.begin(), permutation.end(), 0); std::iota(permutation.begin(), permutation.end(), 0);
if(c_ins == b_ins) if(c_ins == b_ins)
...@@ -310,7 +310,8 @@ struct find_dot_mul ...@@ -310,7 +310,8 @@ struct find_dot_mul
broadcast_v["out_lens"] = c_lens; broadcast_v["out_lens"] = c_lens;
auto db_ins = auto db_ins =
m.insert_instruction(ins, make_op(d_ins->name(), broadcast_v), d_ins->inputs()); m.insert_instruction(ins, make_op(d_ins->name(), broadcast_v), d_ins->inputs());
auto db_transpose_ins = m.insert_instruction(ins, make_op("transpose", {{"permutation", permutation}}), db_ins); auto db_transpose_ins =
m.insert_instruction(ins, make_op("transpose", {{"permutation", permutation}}), db_ins);
auto cd_ins = m.insert_instruction(ins, make_op("mul"), c_ins, db_transpose_ins); auto cd_ins = m.insert_instruction(ins, make_op("mul"), c_ins, db_transpose_ins);
if(c_ins == b_ins) if(c_ins == b_ins)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment