https://github.com/halide/Halide
Raw File
Tip revision: 13f388d9eb3850d53e5725ea8814f0acce395da6 authored by Andrew Adams on 25 August 2020, 21:15:53 UTC
Merge remote-tracking branch 'origin/master' into abadams/avoid_name_mangling_in_cross_module_dependencies
Tip revision: 13f388d
ScheduleFunctions.h
#ifndef HALIDE_INTERNAL_SCHEDULE_FUNCTIONS_H
#define HALIDE_INTERNAL_SCHEDULE_FUNCTIONS_H

/** \file
 *
 * Defines the function that does initial lowering of Halide Functions
 * into a loop nest using its schedule. The first stage of lowering.
 */

#include <map>
#include <string>
#include <vector>

#include "Expr.h"
#include "Target.h"

namespace Halide {
namespace Internal {

class Function;

/** Build loop nests and inject Function realizations at the
 * appropriate places using the schedule. Returns a flag indicating
 * whether memoization passes need to be run. */
Stmt schedule_functions(const std::vector<Function> &outputs,
                        const std::vector<std::vector<std::string>> &fused_groups,
                        const std::map<std::string, Function> &env,
                        const Target &target,
                        bool &any_memoized);

}  // namespace Internal
}  // namespace Halide

#endif
back to top