Commit 13238ba1 authored by Paul's avatar Paul
Browse files

Format

parent eb186040
...@@ -48,8 +48,7 @@ struct precompile_op ...@@ -48,8 +48,7 @@ struct precompile_op
{ {
return pack(f(self.op, "op"), return pack(f(self.op, "op"),
f(self.additional_args, "additional_args"), f(self.additional_args, "additional_args"),
f(self.ignore_modules, "ignore_modules") f(self.ignore_modules, "ignore_modules"));
);
} }
std::string name() const { return "gpu::precompile_op"; } std::string name() const { return "gpu::precompile_op"; }
...@@ -58,7 +57,7 @@ struct precompile_op ...@@ -58,7 +57,7 @@ struct precompile_op
{ {
// Pop off additional args // Pop off additional args
inputs.resize(inputs.size() - additional_args); inputs.resize(inputs.size() - additional_args);
if (ignore_modules) if(ignore_modules)
return op.compute_shape(inputs); return op.compute_shape(inputs);
return op.compute_shape(inputs, mods); return op.compute_shape(inputs, mods);
} }
......
...@@ -1190,8 +1190,8 @@ struct find_concat_pointwise ...@@ -1190,8 +1190,8 @@ struct find_concat_pointwise
{ {
auto matcher() const auto matcher() const
{ {
return precompile_name("pointwise")(match::arg(0)( return precompile_name("pointwise")(
precompile_name("concat").bind("concat"))); match::arg(0)(precompile_name("concat").bind("concat")));
} }
void apply(module& m, const match::matcher_result& r) const void apply(module& m, const match::matcher_result& r) const
......
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