write_literals.cpp

#include <migraph/gpu/write_literals.hpp>
#include <migraph/iterator_for.hpp>
#include <migraph/gpu/hip.hpp>
#include <migraph/instruction.hpp>

namespace migraph {

namespace gpu {

void write_literals::apply(program& p) const
{
    for(auto ins : iterator_for(p))
    {
        if(ins->op.name() == "@literal")
        {
            literal l = ins->lit;
            auto pre  = p.add_literal(l);
            p.replace_instruction(ins, hip_write{}, pre);
        }
    }
}

} // namespace gpu

} // namespace migraph