Commit b1ba0937 authored by Paul's avatar Paul
Browse files

Rename lit field

parent 62970436
......@@ -160,7 +160,7 @@ struct onnx_parser
}
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)); });
}
return prog.add_instruction(op, args[0]);
......
......@@ -71,7 +71,7 @@ void verify_instructions(const migraph::program& prog, double tolerance = 80)
for(auto&& arg : ins.inputs())
{
if(arg->name() == "@literal")
inputs.push_back(p.add_literal(arg->lit));
inputs.push_back(p.add_literal(arg->get_literal()));
else
inputs.push_back(
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)
if(ins->name() == "@literal")
{
if(ins->lit.get_shape().elements() > 10)
if(ins->get_literal().get_shape().elements() > 10)
os << "{ ... }";
else
os << "{" << ins->lit << "}";
os << "{" << ins->get_literal() << "}";
}
if(!ins->inputs().empty())
......@@ -286,7 +286,7 @@ argument generic_eval(const program& p,
{
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")
{
......
......@@ -30,7 +30,7 @@ void write_literals::apply(program& p) const
{
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();
ctx->literals.push_back(a);
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