Commit e2967e04 authored by Paul's avatar Paul
Browse files

Format

parent df3749cd
......@@ -14,7 +14,10 @@ namespace gpu {
std::string dump_mlir(const module& m);
code_object_op compile_mlir(const module& m);
instruction_ref insert_mlir(module& m, instruction_ref ins, const module& mmlir, const std::vector<instruction_ref>& inputs);
instruction_ref insert_mlir(module& m,
instruction_ref ins,
const module& mmlir,
const std::vector<instruction_ref>& inputs);
} // namespace gpu
} // namespace MIGRAPHX_INLINE_NS
......
......@@ -510,7 +510,10 @@ code_object_op compile_mlir(const module& m)
return mp.compile();
}
instruction_ref insert_mlir(module& m, instruction_ref ins, const module& mmlir, const std::vector<instruction_ref>& inputs)
instruction_ref insert_mlir(module& m,
instruction_ref ins,
const module& mmlir,
const std::vector<instruction_ref>& inputs)
{
auto co = compile_mlir(mmlir);
......@@ -527,10 +530,10 @@ instruction_ref insert_mlir(module& m, instruction_ref ins, const module& mmlir,
return lit;
};
for(auto input:inputs)
for(auto input : inputs)
{
const size_t offset = 0;
auto s = input->get_shape();
auto s = input->get_shape();
refs.push_back(input);
refs.push_back(input);
refs.push_back(get_literal(offset)); // offset
......@@ -550,7 +553,7 @@ instruction_ref insert_mlir(module& m, instruction_ref ins, const module& mmlir,
refs.push_back(get_literal(1)); // G
}
co.expected_inputs = to_shapes(refs);
co.output = mmlir.get_output_shapes().front();
co.output = mmlir.get_output_shapes().front();
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