Revision 3fbb12aa928a87a4c4e69a95eaf6f5ee02958600 authored by John Lawson on 09 February 2021, 18:00:02 UTC, committed by GitHub on 09 February 2021, 18:00:02 UTC
* Add support for AVX512 BF16 dot product * Match on f32*f32 * Remove f32 check
1 parent 3e034d6
Lower.h
#ifndef HALIDE_INTERNAL_LOWER_H
#define HALIDE_INTERNAL_LOWER_H
/** \file
*
* Defines the function that generates a statement that computes a
* Halide function using its schedule.
*/
#include <string>
#include <vector>
#include "Argument.h"
#include "Expr.h"
#include "Module.h"
namespace Halide {
struct Target;
namespace Internal {
class Function;
class IRMutator;
/** Given a vector of scheduled halide functions, create a Module that
* evaluates it. Automatically pulls in all the functions f depends
* on. Some stages of lowering may be target-specific. The Module may
* contain submodules for computation offloaded to another execution
* engine or API as well as buffers that are used in the passed in
* Stmt. */
Module lower(const std::vector<Function> &output_funcs,
const std::string &pipeline_name,
const Target &t,
const std::vector<Argument> &args,
LinkageType linkage_type,
const std::vector<Stmt> &requirements = std::vector<Stmt>(),
bool trace_pipeline = false,
const std::vector<IRMutator *> &custom_passes = std::vector<IRMutator *>());
/** Given a halide function with a schedule, create a statement that
* evaluates it. Automatically pulls in all the functions f depends
* on. Some stages of lowering may be target-specific. Mostly used as
* a convenience function in tests that wish to assert some property
* of the lowered IR. */
Stmt lower_main_stmt(const std::vector<Function> &output_funcs,
const std::string &pipeline_name,
const Target &t,
const std::vector<Stmt> &requirements = std::vector<Stmt>(),
bool trace_pipeline = false,
const std::vector<IRMutator *> &custom_passes = std::vector<IRMutator *>());
void lower_test();
} // namespace Internal
} // namespace Halide
#endif
Computing file changes ...