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

Format

parent eb186040
......@@ -41,15 +41,14 @@ struct precompile_op
{
operation op = op::identity{};
std::size_t additional_args = 1;
bool ignore_modules = false;
bool ignore_modules = false;
template <class Self, class F>
static auto reflect(Self& self, F f)
{
return pack(f(self.op, "op"),
f(self.additional_args, "additional_args"),
f(self.ignore_modules, "ignore_modules")
);
return pack(f(self.op, "op"),
f(self.additional_args, "additional_args"),
f(self.ignore_modules, "ignore_modules"));
}
std::string name() const { return "gpu::precompile_op"; }
......@@ -58,7 +57,7 @@ struct precompile_op
{
// Pop off additional args
inputs.resize(inputs.size() - additional_args);
if (ignore_modules)
if(ignore_modules)
return op.compute_shape(inputs);
return op.compute_shape(inputs, mods);
}
......
......@@ -1190,15 +1190,15 @@ struct find_concat_pointwise
{
auto matcher() const
{
return precompile_name("pointwise")(match::arg(0)(
precompile_name("concat").bind("concat")));
return precompile_name("pointwise")(
match::arg(0)(precompile_name("concat").bind("concat")));
}
void apply(module& m, const match::matcher_result& r) const
{
auto ins = r.result;
auto ins = r.result;
auto concat = r.instructions["concat"];
auto* pm = ins->module_inputs().front();
auto* pm = ins->module_inputs().front();
if(not concat->module_inputs().empty())
return;
......
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