1 #ifndef HALIDE_CODEGEN_INTERNAL_H
2 #define HALIDE_CODEGEN_INTERNAL_H
22 class IRBuilderDefaultInserter;
30 template<
typename,
typename>
96 void get_target_options(
const llvm::Module &module, llvm::TargetOptions &options, std::string &mcpu, std::string &mattrs);
Base classes for Halide expressions (Halide::Expr) and statements (Halide::Internal::Stmt)
Defines the Scope class, which is used for keeping track of names in a scope while traversing IR.
void set_function_attributes_for_target(llvm::Function *, const Target &)
Set the appropriate llvm Function attributes given a Target.
std::pair< Expr, Expr > long_div_mod_round_to_zero(const Expr &a, const Expr &b, const uint64_t *max_abs=nullptr)
Does a {div/mod}_round_to_zero using binary long division for int/uint.
void clone_target_options(const llvm::Module &from, llvm::Module &to)
Given two llvm::Modules, clone target options from one to the other.
llvm::Type * get_vector_element_type(llvm::Type *)
Get the scalar type of an llvm vector type.
Expr lower_int_uint_div(const Expr &a, const Expr &b, bool round_to_zero=false)
Given a Halide Euclidean division/mod operation, do constant optimizations and possibly call lower_eu...
Expr lower_signed_shift_right(const Expr &a, const Expr &b)
Given a Halide shift operation with a signed shift amount (may be negative), define an equivalent exp...
llvm::Type * get_vector_type(llvm::Type *, int)
void embed_bitcode(llvm::Module *M, const std::string &halide_command)
Save a copy of the llvm IR currently represented by the module as data in the __LLVM,...
Expr lower_int_uint_mod(const Expr &a, const Expr &b)
Given a Halide Euclidean division/mod operation, do constant optimizations and possibly call lower_eu...
bool can_allocation_fit_on_stack(int64_t size)
Given a size (in bytes), return True if the allocation size can fit on the stack; otherwise,...
llvm::Type * llvm_type_of(llvm::LLVMContext *context, Halide::Type t)
Get the llvm type equivalent to a given halide type.
llvm::ElementCount element_count(int e)
Expr lower_mux(const Call *mux)
Reduce a mux intrinsic to a select tree.
void get_target_options(const llvm::Module &module, llvm::TargetOptions &options, std::string &mcpu, std::string &mattrs)
Given an llvm::Module, set llvm:TargetOptions, cpu and attr information.
Expr lower_signed_shift_left(const Expr &a, const Expr &b)
Given a Halide shift operation with a signed shift amount (may be negative), define an equivalent exp...
Expr lower_euclidean_div(Expr a, Expr b)
Given a Halide Euclidean division/mod operation, define it in terms of div_round_to_zero or mod_round...
int get_vector_num_elements(llvm::Type *)
Get the number of elements in an llvm vector type, or return 1 if it's not a vector type.
bool function_takes_user_context(const std::string &name)
Which built-in functions require a user-context first argument?
Expr lower_euclidean_mod(Expr a, Expr b)
Given a Halide Euclidean division/mod operation, define it in terms of div_round_to_zero or mod_round...
std::unique_ptr< llvm::TargetMachine > make_target_machine(const llvm::Module &module)
Given an llvm::Module, get or create an llvm:TargetMachine.
This file defines the class FunctionDAG, which is our representation of a Halide pipeline,...
Expr mux(const Expr &id, const std::initializer_list< Expr > &values)
Oftentimes we want to pack a list of expressions with the same type into a channel dimension,...
unsigned __INT64_TYPE__ uint64_t
signed __INT64_TYPE__ int64_t
A fragment of Halide syntax.
A struct representing a target machine and os to generate code for.
Types in the halide type system.