"scripts/convert_yi_vl.py" did not exist on "79cb018e4bcabe9d22db64657649cc1930370a3d"
Commit 61a1c018 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

add a function to quantize a specific instruction

parent 5482a441
#ifndef MIGRAPHX_GUARD_RTGLIB_QUANTIZE_INS_HPP
#define MIGRAPHX_GUARD_RTGLIB_QUANTIZE_INS_HPP
#include <string>
#include <vector>
#include <migraphx/instruction_ref.hpp>
#include <migraphx/operation.hpp>
#include <migraphx/config.hpp>
namespace migraphx {
inline namespace MIGRAPHX_INLINE_NS {
struct program;
void quantize_ins(program& prog, const std::vector<std::string>& ins_names);
} // namespace MIGRAPHX_INLINE_NS
} // namespace migraphx
#endif
#include <migraphx/quantize_ins.hpp>
#include <migraphx/program.hpp>
#include <migraphx/instruction.hpp>
#include <migraphx/iterator_for.hpp>
#include <migraphx/stringutils.hpp>
#include <utility>
namespace migraphx {
inline namespace MIGRAPHX_INLINE_NS {
void quantize_ins(program& prog, const std::vector<std::string>& ins_names)
{
for (auto ins : iterator_for(prog))
{
auto name_it = std::find(ins_name.begin(), ins_name.end(), ins->name());
if (name_it == ins_name.end())
{
continue;
}
}
}
} // namespace MIGRAPHX_INLINE_NS
} // namespace migraphx
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