void populate_environment(Function f, map<string, Function> &env, bool recursive = true) {
map<string, Function>::const_iterator iter = env.find(f.name());
if (iter != env.end()) {
assert(iter->second.same_as(f) &&
"Can't compile a pipeline using multiple functions with same name");
return;
}
FindCalls calls;
for (size_t i = 0; i < f.values().size(); i++) {
f.values()[i].accept(&calls);
}
// Consider reductions
for (size_t j = 0; j < f.reductions().size(); j++) {
ReductionDefinition r = f.reductions()[j];
for (size_t i = 0; i < r.values.size(); i++) {
r.values[i].accept(&calls);
}
for (size_t i = 0; i < r.args.size(); i++) {
r.args[i].accept(&calls);
}
ReductionDomain d = r.domain;
if (r.domain.defined()) {
for (size_t i = 0; i < d.domain().size(); i++) {
d.domain()[i].min.accept(&calls);
d.domain()[i].extent.accept(&calls);
}
}
}
// Consider extern calls
if (f.has_extern_definition()) {
for (size_t i = 0; i < f.extern_arguments().size(); i++) {
ExternFuncArgument arg = f.extern_arguments()[i];
if (arg.is_func()) {
Function g(arg.func);
calls.calls[g.name()] = g;
}
}
}
if (!recursive) {
env.insert(calls.calls.begin(), calls.calls.end());
} else {
env[f.name()] = f;
for (map<string, Function>::const_iterator iter = calls.calls.begin();
iter != calls.calls.end(); ++iter) {
populate_environment(iter->second, env);
}
}
}
void IRVisitor::visit(const Call *op) {
for (size_t i = 0; i < op->args.size(); i++) {
op->args[i].accept(this);
}
// Consider extern call args
Function f = op->func;
if (op->call_type == Call::Halide && f.has_extern_definition()) {
for (size_t i = 0; i < f.extern_arguments().size(); i++) {
ExternFuncArgument arg = f.extern_arguments()[i];
if (arg.is_expr()) {
arg.expr.accept(this);
}
}
}
}
AllocationInference(const map<string, Function> &e, const FuncValueBounds &fb) :
env(e), func_bounds(fb) {
// Figure out which buffers are touched by extern stages
for (map<string, Function>::const_iterator iter = e.begin();
iter != e.end(); ++iter) {
Function f = iter->second;
if (f.has_extern_definition()) {
touched_by_extern.insert(f.name());
for (size_t i = 0; i < f.extern_arguments().size(); i++) {
ExternFuncArgument arg = f.extern_arguments()[i];
if (!arg.is_func()) continue;
Function input(arg.func);
touched_by_extern.insert(input.name());
}
}
}
}
