Halide 19.0.0
Halide compiler and libraries
Loading...
Searching...
No Matches
Public Member Functions | Protected Member Functions | List of all members
Halide::Internal::IRVisitor Class Reference

A base class for algorithms that need to recursively walk over the IR. More...

#include <IRVisitor.h>

Inheritance diagram for Halide::Internal::IRVisitor:
Halide::Internal::Closure Halide::Internal::CodeGen_LLVM Halide::Internal::FindAllCalls Halide::Internal::IRGraphVisitor Halide::Internal::IRPrinter Halide::Internal::HostClosure Halide::Internal::CodeGen_Posix Halide::Internal::ExprUsesVars< T > Halide::Internal::CodeGen_C Halide::Internal::CodeGen_PyTorch Halide::Internal::CodeGen_GPU_C

Public Member Functions

 IRVisitor ()=default
 
virtual ~IRVisitor ()=default
 

Protected Member Functions

virtual void visit (const IntImm *)
 
virtual void visit (const UIntImm *)
 
virtual void visit (const FloatImm *)
 
virtual void visit (const StringImm *)
 
virtual void visit (const Cast *)
 
virtual void visit (const Reinterpret *)
 
virtual void visit (const Variable *)
 
virtual void visit (const Add *)
 
virtual void visit (const Sub *)
 
virtual void visit (const Mul *)
 
virtual void visit (const Div *)
 
virtual void visit (const Mod *)
 
virtual void visit (const Min *)
 
virtual void visit (const Max *)
 
virtual void visit (const EQ *)
 
virtual void visit (const NE *)
 
virtual void visit (const LT *)
 
virtual void visit (const LE *)
 
virtual void visit (const GT *)
 
virtual void visit (const GE *)
 
virtual void visit (const And *)
 
virtual void visit (const Or *)
 
virtual void visit (const Not *)
 
virtual void visit (const Select *)
 
virtual void visit (const Load *)
 
virtual void visit (const Ramp *)
 
virtual void visit (const Broadcast *)
 
virtual void visit (const Call *)
 
virtual void visit (const Let *)
 
virtual void visit (const LetStmt *)
 
virtual void visit (const AssertStmt *)
 
virtual void visit (const ProducerConsumer *)
 
virtual void visit (const For *)
 
virtual void visit (const Store *)
 
virtual void visit (const Provide *)
 
virtual void visit (const Allocate *)
 
virtual void visit (const Free *)
 
virtual void visit (const Realize *)
 
virtual void visit (const Block *)
 
virtual void visit (const IfThenElse *)
 
virtual void visit (const Evaluate *)
 
virtual void visit (const Shuffle *)
 
virtual void visit (const VectorReduce *)
 
virtual void visit (const Prefetch *)
 
virtual void visit (const Fork *)
 
virtual void visit (const Acquire *)
 
virtual void visit (const Atomic *)
 
virtual void visit (const HoistedStorage *)
 

Detailed Description

A base class for algorithms that need to recursively walk over the IR.

The default implementations just recursively walk over the children. Override the ones you care about.

Definition at line 19 of file IRVisitor.h.

Constructor & Destructor Documentation

◆ IRVisitor()

Halide::Internal::IRVisitor::IRVisitor ( )
default

◆ ~IRVisitor()

virtual Halide::Internal::IRVisitor::~IRVisitor ( )
virtualdefault

Member Function Documentation

◆ visit() [1/48]

virtual void Halide::Internal::IRVisitor::visit ( const IntImm * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

Referenced by Halide::SimdOpCheckTest::check_one().

◆ visit() [2/48]

virtual void Halide::Internal::IRVisitor::visit ( const UIntImm * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [3/48]

virtual void Halide::Internal::IRVisitor::visit ( const FloatImm * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [4/48]

virtual void Halide::Internal::IRVisitor::visit ( const StringImm * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [5/48]

virtual void Halide::Internal::IRVisitor::visit ( const Cast * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [6/48]

virtual void Halide::Internal::IRVisitor::visit ( const Reinterpret * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [7/48]

virtual void Halide::Internal::IRVisitor::visit ( const Variable * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [8/48]

virtual void Halide::Internal::IRVisitor::visit ( const Add * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [9/48]

virtual void Halide::Internal::IRVisitor::visit ( const Sub * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [10/48]

virtual void Halide::Internal::IRVisitor::visit ( const Mul * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [11/48]

virtual void Halide::Internal::IRVisitor::visit ( const Div * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [12/48]

virtual void Halide::Internal::IRVisitor::visit ( const Mod * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [13/48]

virtual void Halide::Internal::IRVisitor::visit ( const Min * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [14/48]

virtual void Halide::Internal::IRVisitor::visit ( const Max * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [15/48]

virtual void Halide::Internal::IRVisitor::visit ( const EQ * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [16/48]

virtual void Halide::Internal::IRVisitor::visit ( const NE * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [17/48]

virtual void Halide::Internal::IRVisitor::visit ( const LT * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [18/48]

virtual void Halide::Internal::IRVisitor::visit ( const LE * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [19/48]

virtual void Halide::Internal::IRVisitor::visit ( const GT * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [20/48]

virtual void Halide::Internal::IRVisitor::visit ( const GE * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [21/48]

virtual void Halide::Internal::IRVisitor::visit ( const And * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [22/48]

virtual void Halide::Internal::IRVisitor::visit ( const Or * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [23/48]

virtual void Halide::Internal::IRVisitor::visit ( const Not * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [24/48]

virtual void Halide::Internal::IRVisitor::visit ( const Select * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [25/48]

virtual void Halide::Internal::IRVisitor::visit ( const Load * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [26/48]

virtual void Halide::Internal::IRVisitor::visit ( const Ramp * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [27/48]

virtual void Halide::Internal::IRVisitor::visit ( const Broadcast * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [28/48]

virtual void Halide::Internal::IRVisitor::visit ( const Call * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::HostClosure, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [29/48]

virtual void Halide::Internal::IRVisitor::visit ( const Let * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [30/48]

virtual void Halide::Internal::IRVisitor::visit ( const LetStmt * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [31/48]

virtual void Halide::Internal::IRVisitor::visit ( const AssertStmt * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [32/48]

virtual void Halide::Internal::IRVisitor::visit ( const ProducerConsumer * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [33/48]

virtual void Halide::Internal::IRVisitor::visit ( const For * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::HostClosure, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [34/48]

virtual void Halide::Internal::IRVisitor::visit ( const Store * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [35/48]

virtual void Halide::Internal::IRVisitor::visit ( const Provide * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [36/48]

virtual void Halide::Internal::IRVisitor::visit ( const Allocate * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [37/48]

virtual void Halide::Internal::IRVisitor::visit ( const Free * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [38/48]

virtual void Halide::Internal::IRVisitor::visit ( const Realize * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [39/48]

virtual void Halide::Internal::IRVisitor::visit ( const Block * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [40/48]

virtual void Halide::Internal::IRVisitor::visit ( const IfThenElse * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [41/48]

virtual void Halide::Internal::IRVisitor::visit ( const Evaluate * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [42/48]

virtual void Halide::Internal::IRVisitor::visit ( const Shuffle * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [43/48]

virtual void Halide::Internal::IRVisitor::visit ( const VectorReduce * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [44/48]

virtual void Halide::Internal::IRVisitor::visit ( const Prefetch * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [45/48]

virtual void Halide::Internal::IRVisitor::visit ( const Fork * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [46/48]

virtual void Halide::Internal::IRVisitor::visit ( const Acquire * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [47/48]

virtual void Halide::Internal::IRVisitor::visit ( const Atomic * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

◆ visit() [48/48]

virtual void Halide::Internal::IRVisitor::visit ( const HoistedStorage * )
protectedvirtual

Reimplemented in Halide::Internal::Closure, Halide::Internal::CodeGen_C, Halide::Internal::CodeGen_GPU_C, Halide::Internal::CodeGen_LLVM, Halide::Internal::CodeGen_Posix, Halide::Internal::IRGraphVisitor, and Halide::Internal::IRPrinter.

The documentation for this class was generated from the following file: