Halide 19.0.0
Halide compiler and libraries
|
This class emits C++ code to wrap a Halide pipeline so that it can be used as a C++ extension operator in PyTorch. More...
#include <CodeGen_PyTorch.h>
Public Member Functions | |
CodeGen_PyTorch (std::ostream &dest) | |
~CodeGen_PyTorch () override=default | |
void | compile (const Module &module) |
Emit the PyTorch C++ wrapper for the Halide pipeline. | |
Public Member Functions inherited from Halide::Internal::IRPrinter | |
IRPrinter (std::ostream &) | |
Construct an IRPrinter pointed at a given output stream (e.g. | |
void | print (const Expr &) |
emit an expression on the output stream | |
void | print_no_parens (const Expr &) |
Emit an expression on the output stream without enclosing parens. | |
void | print (const Stmt &) |
emit a statement on the output stream | |
void | print_list (const std::vector< Expr > &exprs) |
emit a comma delimited list of exprs, without any leading or trailing punctuation. | |
Public Member Functions inherited from Halide::Internal::IRVisitor | |
IRVisitor ()=default | |
virtual | ~IRVisitor ()=default |
Additional Inherited Members | |
Static Public Member Functions inherited from Halide::Internal::IRPrinter | |
static void | test () |
Protected Member Functions inherited from Halide::Internal::IRPrinter | |
Indentation | get_indent () const |
void | open () |
Either emits "(" or "", depending on the value of implicit_parens. | |
void | close () |
Either emits ")" or "", depending on the value of implicit_parens. | |
void | print_lets (const Let *let) |
A helper for printing a chain of lets with line breaks. | |
void | visit (const IntImm *) override |
void | visit (const UIntImm *) override |
void | visit (const FloatImm *) override |
void | visit (const StringImm *) override |
void | visit (const Cast *) override |
void | visit (const Reinterpret *) override |
void | visit (const Variable *) override |
void | visit (const Add *) override |
void | visit (const Sub *) override |
void | visit (const Mul *) override |
void | visit (const Div *) override |
void | visit (const Mod *) override |
void | visit (const Min *) override |
void | visit (const Max *) override |
void | visit (const EQ *) override |
void | visit (const NE *) override |
void | visit (const LT *) override |
void | visit (const LE *) override |
void | visit (const GT *) override |
void | visit (const GE *) override |
void | visit (const And *) override |
void | visit (const Or *) override |
void | visit (const Not *) override |
void | visit (const Select *) override |
void | visit (const Load *) override |
void | visit (const Ramp *) override |
void | visit (const Broadcast *) override |
void | visit (const Call *) override |
void | visit (const Let *) override |
void | visit (const LetStmt *) override |
void | visit (const AssertStmt *) override |
void | visit (const ProducerConsumer *) override |
void | visit (const For *) override |
void | visit (const Acquire *) override |
void | visit (const Store *) override |
void | visit (const Provide *) override |
void | visit (const Allocate *) override |
void | visit (const Free *) override |
void | visit (const Realize *) override |
void | visit (const Block *) override |
void | visit (const Fork *) override |
void | visit (const IfThenElse *) override |
void | visit (const Evaluate *) override |
void | visit (const Shuffle *) override |
void | visit (const VectorReduce *) override |
void | visit (const Prefetch *) override |
void | visit (const Atomic *) override |
void | visit (const HoistedStorage *) override |
Protected Member Functions inherited from Halide::Internal::IRVisitor | |
Protected Attributes inherited from Halide::Internal::IRPrinter | |
std::ostream & | stream |
The stream on which we're outputting. | |
int | indent = 0 |
The current indentation level, useful for pretty-printing statements. | |
bool | implicit_parens = false |
Certain expressions do not need parens around them, e.g. | |
Scope | known_type |
The symbols whose types can be inferred from values printed already. | |
This class emits C++ code to wrap a Halide pipeline so that it can be used as a C++ extension operator in PyTorch.
Definition at line 29 of file CodeGen_PyTorch.h.
Halide::Internal::CodeGen_PyTorch::CodeGen_PyTorch | ( | std::ostream & | dest | ) |
|
overridedefault |
void Halide::Internal::CodeGen_PyTorch::compile | ( | const Module & | module | ) |