Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "1df572dc9784c67c739ac09a3f1179304145df1e"
Commit 467a7cb8 authored by Paul's avatar Paul
Browse files

Format

parent 88f549e2
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/targets/gpu/mlir.cpp
View file @ 467a7cb8
...@@ -539,7 +539,7 @@ instruction_ref insert_mlir(module& m, ...@@ -539,7 +539,7 @@ instruction_ref insert_mlir(module& m,
{ {
const size_t offset = 0; const size_t offset = 0;
auto s = input->get_shape(); auto s = input->get_shape();
last = refs.size(); last = refs.size();
refs.push_back(input); refs.push_back(input);
refs.push_back(input); refs.push_back(input);
refs.push_back(get_literal(offset)); // offset refs.push_back(get_literal(offset)); // offset
...@@ -560,7 +560,7 @@ instruction_ref insert_mlir(module& m, ...@@ -560,7 +560,7 @@ instruction_ref insert_mlir(module& m,
} }
co.expected_inputs = to_shapes(refs); co.expected_inputs = to_shapes(refs);
co.output = mmlir.get_output_shapes().front(); co.output = mmlir.get_output_shapes().front();
co.output_arg = last; co.output_arg = last;
return m.insert_instruction(ins, co, refs); return m.insert_instruction(ins, co, refs);
} }
......
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