Commit 96513f1d authored by Paul's avatar Paul
Browse files

Formatting

parent e637b5aa
...@@ -14,7 +14,7 @@ auto get_hash(const T& x) ...@@ -14,7 +14,7 @@ auto get_hash(const T& x)
return std::hash<T>{}(x); return std::hash<T>{}(x);
} }
template<class F> template <class F>
migraph::argument run_cpu(F f) migraph::argument run_cpu(F f)
{ {
auto p = f(); auto p = f();
...@@ -29,7 +29,7 @@ migraph::argument run_cpu(F f) ...@@ -29,7 +29,7 @@ migraph::argument run_cpu(F f)
return out; return out;
} }
template<class F> template <class F>
migraph::argument run_gpu(F f) migraph::argument run_gpu(F f)
{ {
auto p = f(); auto p = f();
...@@ -45,8 +45,8 @@ migraph::argument run_gpu(F f) ...@@ -45,8 +45,8 @@ migraph::argument run_gpu(F f)
return migraph::gpu::from_gpu(out); return migraph::gpu::from_gpu(out);
} }
template<class F> template <class F>
void verify_program(const std::string & name, F f, double tolerance = 100) void verify_program(const std::string& name, F f, double tolerance = 100)
{ {
auto x = run_cpu(f); auto x = run_cpu(f);
auto y = run_gpu(f); auto y = run_gpu(f);
...@@ -55,7 +55,7 @@ void verify_program(const std::string & name, F f, double tolerance = 100) ...@@ -55,7 +55,7 @@ void verify_program(const std::string & name, F f, double tolerance = 100)
void verify_instructions(const migraph::program& prog, double tolerance = 100) void verify_instructions(const migraph::program& prog, double tolerance = 100)
{ {
for(auto&& ins:prog) for(auto&& ins : prog)
{ {
if(ins.op.name().front() == '@') if(ins.op.name().front() == '@')
continue; continue;
...@@ -68,12 +68,13 @@ void verify_instructions(const migraph::program& prog, double tolerance = 100) ...@@ -68,12 +68,13 @@ void verify_instructions(const migraph::program& prog, double tolerance = 100)
auto create_program = [&] { auto create_program = [&] {
migraph::program p; migraph::program p;
std::vector<migraph::instruction_ref> inputs; std::vector<migraph::instruction_ref> inputs;
for(auto&& arg:ins.arguments) for(auto&& arg : ins.arguments)
{ {
if(arg->op.name() == "@literal") if(arg->op.name() == "@literal")
inputs.push_back(p.add_literal(arg->lit)); inputs.push_back(p.add_literal(arg->lit));
else else
inputs.push_back(p.add_parameter(std::to_string(inputs.size()), arg->get_shape())); inputs.push_back(
p.add_parameter(std::to_string(inputs.size()), arg->get_shape()));
} }
p.add_instruction(ins.op, inputs); p.add_instruction(ins.op, inputs);
return p; return p;
...@@ -94,7 +95,7 @@ void verify_instructions(const migraph::program& prog, double tolerance = 100) ...@@ -94,7 +95,7 @@ void verify_instructions(const migraph::program& prog, double tolerance = 100)
int main(int argc, char const* argv[]) int main(int argc, char const* argv[])
{ {
std::vector<std::string> args(argv+1, argv+argc); std::vector<std::string> args(argv + 1, argv + argc);
if(not args.empty()) if(not args.empty())
{ {
std::string file = args.front(); std::string file = args.front();
......
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