Commit b1ba0937 authored by Paul's avatar Paul
Browse files

Rename lit field

parent 62970436
...@@ -160,7 +160,7 @@ struct onnx_parser ...@@ -160,7 +160,7 @@ struct onnx_parser
} }
if(args.size() == 2) if(args.size() == 2)
{ {
literal s = args[1]->lit; literal s = args[1]->get_literal();
s.visit([&](auto v) { copy(v, std::back_inserter(op.dims)); }); s.visit([&](auto v) { copy(v, std::back_inserter(op.dims)); });
} }
return prog.add_instruction(op, args[0]); return prog.add_instruction(op, args[0]);
......
...@@ -71,7 +71,7 @@ void verify_instructions(const migraph::program& prog, double tolerance = 80) ...@@ -71,7 +71,7 @@ void verify_instructions(const migraph::program& prog, double tolerance = 80)
for(auto&& arg : ins.inputs()) for(auto&& arg : ins.inputs())
{ {
if(arg->name() == "@literal") if(arg->name() == "@literal")
inputs.push_back(p.add_literal(arg->lit)); inputs.push_back(p.add_literal(arg->get_literal()));
else else
inputs.push_back( inputs.push_back(
p.add_parameter(std::to_string(inputs.size()), arg->get_shape())); p.add_parameter(std::to_string(inputs.size()), arg->get_shape()));
......
...@@ -42,10 +42,10 @@ static void print_program(std::ostream& os, const program& p, F annonate) ...@@ -42,10 +42,10 @@ static void print_program(std::ostream& os, const program& p, F annonate)
if(ins->name() == "@literal") if(ins->name() == "@literal")
{ {
if(ins->lit.get_shape().elements() > 10) if(ins->get_literal().get_shape().elements() > 10)
os << "{ ... }"; os << "{ ... }";
else else
os << "{" << ins->lit << "}"; os << "{" << ins->get_literal() << "}";
} }
if(!ins->inputs().empty()) if(!ins->inputs().empty())
...@@ -286,7 +286,7 @@ argument generic_eval(const program& p, ...@@ -286,7 +286,7 @@ argument generic_eval(const program& p,
{ {
if(ins->name() == "@literal") if(ins->name() == "@literal")
{ {
results.emplace(ins, trace(ins, [&] { return ins->lit.get_argument(); })); results.emplace(ins, trace(ins, [&] { return ins->get_literal().get_argument(); }));
} }
else if(ins->name() == "@param") else if(ins->name() == "@param")
{ {
......
...@@ -30,7 +30,7 @@ void write_literals::apply(program& p) const ...@@ -30,7 +30,7 @@ void write_literals::apply(program& p) const
{ {
if(ins->name() == "@literal") if(ins->name() == "@literal")
{ {
argument a = to_gpu(ins->lit.get_argument()); argument a = to_gpu(ins->get_literal().get_argument());
std::size_t n = ctx->literals.size(); std::size_t n = ctx->literals.size();
ctx->literals.push_back(a); ctx->literals.push_back(a);
p.replace_instruction(ins, hip_load_literal{a.get_shape(), n}); p.replace_instruction(ins, hip_load_literal{a.get_shape(), n});
......
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