proteus::MLIRCodeBuilder Class Reference

#include <MLIRCodeBuilder.h>

Inheritance diagram for proteus::MLIRCodeBuilder:

Collaboration diagram for proteus::MLIRCodeBuilder:

Classes
struct	Impl

Public Member Functions
	MLIRCodeBuilder (TargetModelType TM=TargetModelType::HOST)
	~MLIRCodeBuilder () override
	MLIRCodeBuilder (const MLIRCodeBuilder &)=delete
MLIRCodeBuilder &	operator= (const MLIRCodeBuilder &)=delete
TargetModelType	getTargetModel () const override
CodeBuilderKind	getBackendKind () const override
void	print ()
	Dump the MLIR module to llvm::outs().
void	printLLVMIR (llvm::raw_ostream &OS)
std::unique_ptr< llvm::LLVMContext >	takeContext ()
std::unique_ptr< llvm::Module >	takeModule ()
void	setDeviceArch (const std::string &Arch)
IRFunction *	addFunction (const std::string &Name, IRType RetTy, const std::vector< IRType > &ArgTys, bool IsKernel=false) override
void	setFunctionName (IRFunction *F, const std::string &Name) override
	Rename the function identified by F.
IRValue *	getArg (IRFunction *F, size_t Idx) override
	Return the Nth argument of F as an IRValue.
void	beginFunction (IRFunction *F, const char *File, int Line) override
	Set F as the active function and begin IR emission.
void	endFunction () override
void	setInsertPointAtEntry () override
void	clearInsertPoint () override
void	createRetVoid () override
void	createRet (IRValue *V) override
IRValue *	createArith (ArithOp Op, IRValue *LHS, IRValue *RHS, IRType Ty) override
IRValue *	createCast (IRValue *V, IRType FromTy, IRType ToTy) override
IRValue *	getConstantInt (IRType Ty, uint64_t Val) override
IRValue *	getConstantFP (IRType Ty, double Val) override
IRValue *	loadScalar (IRValue *Slot, IRType ValueTy) override
	Load the value stored directly in Slot (scalar alloca).
void	storeScalar (IRValue *Slot, IRValue *Val) override
	Store Val directly into Slot (scalar alloca).
VarAlloc	allocScalar (const std::string &Name, IRType ValueTy) override
void	beginIf (IRValue *Cond, const char *File, int Line) override
void	endIf () override
void	beginFor (IRValue *IterSlot, IRType IterTy, IRValue *InitVal, IRValue *UpperBoundVal, IRValue *IncVal, bool IsSigned, const char *File, int Line, LoopHints Hints={}) override
void	endFor () override
void	beginWhile (std::function< IRValue *()> CondFn, const char *File, int Line) override
void	endWhile () override
IRValue *	createAtomicAdd (IRValue *Addr, IRValue *Val) override
IRValue *	createAtomicSub (IRValue *Addr, IRValue *Val) override
IRValue *	createAtomicMax (IRValue *Addr, IRValue *Val) override
IRValue *	createAtomicMin (IRValue *Addr, IRValue *Val) override
IRValue *	createCmp (CmpOp Op, IRValue *LHS, IRValue *RHS, IRType Ty) override
IRValue *	createAnd (IRValue *LHS, IRValue *RHS) override
IRValue *	createOr (IRValue *LHS, IRValue *RHS) override
IRValue *	createXor (IRValue *LHS, IRValue *RHS) override
IRValue *	createNot (IRValue *Val) override
IRValue *	createLoad (IRType Ty, IRValue *Ptr, const std::string &Name="") override
void	createStore (IRValue *Val, IRValue *Ptr) override
IRValue *	createBitCast (IRValue *V, IRType DestTy) override
IRValue *	createZExt (IRValue *V, IRType DestTy) override
VarAlloc	getElementPtr (IRValue *Base, IRType BaseTy, IRValue *Index, IRType ElemTy) override
VarAlloc	getElementPtr (IRValue *Base, IRType BaseTy, size_t Index, IRType ElemTy) override
IRValue *	createCall (const std::string &FName, IRType RetTy, const std::vector< IRType > &ArgTys, const std::vector< IRValue * > &Args) override
IRValue *	createCall (const std::string &FName, IRType RetTy) override
IRValue *	emitIntrinsic (const std::string &Name, IRType RetTy, const std::vector< IRValue * > &Args) override
	Lower a frontend intrinsic name to backend IR.
IRValue *	emitBuiltin (const std::string &Name, IRType RetTy, const std::vector< IRValue * > &Args) override
IRValue *	loadAddress (IRValue *Slot, IRType AllocTy) override
	Load the pointer stored in Slot (pointer alloca).
void	storeAddress (IRValue *Slot, IRValue *Addr) override
	Store Addr into Slot (pointer alloca).
IRValue *	loadFromPointee (IRValue *Slot, IRType AllocTy, IRType ValueTy) override
	Dereference the pointer stored in Slot, then load the pointee.
void	storeToPointee (IRValue *Slot, IRType AllocTy, IRValue *Val) override
	Dereference the pointer stored in Slot, then store Val to it.
VarAlloc	allocPointer (const std::string &Name, IRType ElemTy, unsigned AddrSpace=0) override
VarAlloc	allocArray (const std::string &Name, AddressSpace AS, IRType ElemTy, size_t NElem) override
Public Member Functions inherited from proteus::CodeBuilder
virtual	~CodeBuilder ()=default

Detailed Description

MLIRCodeBuilder implements the CodeBuilder interface using MLIR dialects. Host mode emits func/arith/memref, while CUDA/HIP mode emits gpu.module with gpu.func kernels.

Constructor & Destructor Documentation

MLIRCodeBuilder() [1/2]

proteus::MLIRCodeBuilder::MLIRCodeBuilder ( TargetModelType  TM = TargetModelType::HOST )

explicit

~MLIRCodeBuilder()

proteus::MLIRCodeBuilder::~MLIRCodeBuilder ( )

overridedefault

MLIRCodeBuilder() [2/2]

proteus::MLIRCodeBuilder::MLIRCodeBuilder ( const MLIRCodeBuilder &  )

delete

Member Function Documentation

addFunction()

IRFunction * proteus::MLIRCodeBuilder::addFunction ( const std::string &  Name, IRType  RetTy, const std::vector< IRType > &  ArgTys, bool  IsKernel = false  )

overridevirtual

Create a function with the given name and signature. Kernel intent is explicit so backends can select the correct IR container up front. Returns an opaque IRFunction handle owned by this builder.

Implements proteus::CodeBuilder.

allocArray()

VarAlloc proteus::MLIRCodeBuilder::allocArray ( const std::string &  Name, AddressSpace  AS, IRType  ElemTy, size_t  NElem  )

overridevirtual

Implements proteus::CodeBuilder.

allocPointer()

VarAlloc proteus::MLIRCodeBuilder::allocPointer ( const std::string &  Name, IRType  ElemTy, unsigned  AddrSpace = 0  )

overridevirtual

Implements proteus::CodeBuilder.

allocScalar()

VarAlloc proteus::MLIRCodeBuilder::allocScalar ( const std::string &  Name, IRType  ValueTy  )

overridevirtual

Implements proteus::CodeBuilder.

beginFor()

void proteus::MLIRCodeBuilder::beginFor ( IRValue *  IterSlot, IRType  IterTy, IRValue *  InitVal, IRValue *  UpperBoundVal, IRValue *  IncVal, bool  IsSigned, const char *  File, int  Line, LoopHints  Hints = {}  )

overridevirtual

IterSlot : alloca holding the loop iterator. IterTy : value type of the iterator (must be an integer type). InitVal : initial value to store into IterSlot. UpperBoundVal : exclusive upper bound for the loop condition. IncVal : increment added to the iterator on each iteration. IsSigned : true → ICmpSLT, false → ICmpULT.

Implements proteus::CodeBuilder.

beginFunction()

void proteus::MLIRCodeBuilder::beginFunction ( IRFunction *  F, const char *  File, int  Line  )

overridevirtual

Set F as the active function and begin IR emission.

Implements proteus::CodeBuilder.

beginIf()

void proteus::MLIRCodeBuilder::beginIf ( IRValue *  Cond, const char *  File, int  Line  )

overridevirtual

Implements proteus::CodeBuilder.

beginWhile()

void proteus::MLIRCodeBuilder::beginWhile ( std::function< IRValue *()>  CondFn, const char *  File, int  Line  )

overridevirtual

CondFn : callable that emits the condition IR at the current insert point and returns the resulting i1 Value (true → continue loop).

Implements proteus::CodeBuilder.

clearInsertPoint()

void proteus::MLIRCodeBuilder::clearInsertPoint ( )

overridevirtual

Implements proteus::CodeBuilder.

createAnd()

IRValue * proteus::MLIRCodeBuilder::createAnd ( IRValue *  LHS, IRValue *  RHS  )

overridevirtual

Implements proteus::CodeBuilder.

createArith()

IRValue * proteus::MLIRCodeBuilder::createArith ( ArithOp  Op, IRValue *  LHS, IRValue *  RHS, IRType  Ty  )

overridevirtual

Implements proteus::CodeBuilder.

createAtomicAdd()

IRValue * proteus::MLIRCodeBuilder::createAtomicAdd ( IRValue *  Addr, IRValue *  Val  )

overridevirtual

Implements proteus::CodeBuilder.

createAtomicMax()

IRValue * proteus::MLIRCodeBuilder::createAtomicMax ( IRValue *  Addr, IRValue *  Val  )

overridevirtual

Implements proteus::CodeBuilder.

createAtomicMin()

IRValue * proteus::MLIRCodeBuilder::createAtomicMin ( IRValue *  Addr, IRValue *  Val  )

overridevirtual

Implements proteus::CodeBuilder.

createAtomicSub()

IRValue * proteus::MLIRCodeBuilder::createAtomicSub ( IRValue *  Addr, IRValue *  Val  )

overridevirtual

Implements proteus::CodeBuilder.

createBitCast()

IRValue * proteus::MLIRCodeBuilder::createBitCast ( IRValue *  V, IRType  DestTy  )

overridevirtual

Implements proteus::CodeBuilder.

createCall() [1/2]

IRValue * proteus::MLIRCodeBuilder::createCall ( const std::string &  FName, IRType  RetTy  )

overridevirtual

Implements proteus::CodeBuilder.

createCall() [2/2]

IRValue * proteus::MLIRCodeBuilder::createCall ( const std::string &  FName, IRType  RetTy, const std::vector< IRType > &  ArgTys, const std::vector< IRValue * > &  Args  )

overridevirtual

Implements proteus::CodeBuilder.

createCast()

IRValue * proteus::MLIRCodeBuilder::createCast ( IRValue *  V, IRType  FromTy, IRType  ToTy  )

overridevirtual

Implements proteus::CodeBuilder.

createCmp()

IRValue * proteus::MLIRCodeBuilder::createCmp ( CmpOp  Op, IRValue *  LHS, IRValue *  RHS, IRType  Ty  )

overridevirtual

Implements proteus::CodeBuilder.

createLoad()

IRValue * proteus::MLIRCodeBuilder::createLoad ( IRType  Ty, IRValue *  Ptr, const std::string &  Name = ""  )

overridevirtual

Implements proteus::CodeBuilder.

createNot()

IRValue * proteus::MLIRCodeBuilder::createNot ( IRValue *  Val )

overridevirtual

Implements proteus::CodeBuilder.

createOr()

IRValue * proteus::MLIRCodeBuilder::createOr ( IRValue *  LHS, IRValue *  RHS  )

overridevirtual

Implements proteus::CodeBuilder.

createRet()

void proteus::MLIRCodeBuilder::createRet ( IRValue *  V )

overridevirtual

Implements proteus::CodeBuilder.

createRetVoid()

void proteus::MLIRCodeBuilder::createRetVoid ( )

overridevirtual

Implements proteus::CodeBuilder.

createStore()

void proteus::MLIRCodeBuilder::createStore ( IRValue *  Val, IRValue *  Ptr  )

overridevirtual

Implements proteus::CodeBuilder.

createXor()

IRValue * proteus::MLIRCodeBuilder::createXor ( IRValue *  LHS, IRValue *  RHS  )

overridevirtual

Implements proteus::CodeBuilder.

createZExt()

IRValue * proteus::MLIRCodeBuilder::createZExt ( IRValue *  V, IRType  DestTy  )

overridevirtual

Implements proteus::CodeBuilder.

emitBuiltin()

IRValue * proteus::MLIRCodeBuilder::emitBuiltin ( const std::string &  Name, IRType  RetTy, const std::vector< IRValue * > &  Args  )

overridevirtual

Lower a frontend GPU builtin name to backend IR. Returns nullptr for void builtins (e.g. syncThreads).

Implements proteus::CodeBuilder.

emitIntrinsic()

IRValue * proteus::MLIRCodeBuilder::emitIntrinsic ( const std::string &  Name, IRType  RetTy, const std::vector< IRValue * > &  Args  )

overridevirtual

Lower a frontend intrinsic name to backend IR.

Implements proteus::CodeBuilder.

endFor()

void proteus::MLIRCodeBuilder::endFor ( )

overridevirtual

Implements proteus::CodeBuilder.

endFunction()

void proteus::MLIRCodeBuilder::endFunction ( )

overridevirtual

Implements proteus::CodeBuilder.

endIf()

void proteus::MLIRCodeBuilder::endIf ( )

overridevirtual

Implements proteus::CodeBuilder.

endWhile()

void proteus::MLIRCodeBuilder::endWhile ( )

overridevirtual

Implements proteus::CodeBuilder.

getArg()

IRValue * proteus::MLIRCodeBuilder::getArg ( IRFunction *  F, size_t  Idx  )

overridevirtual

Return the Nth argument of F as an IRValue.

Implements proteus::CodeBuilder.

getBackendKind()

CodeBuilderKind proteus::MLIRCodeBuilder::getBackendKind ( ) const

inlineoverridevirtual

Implements proteus::CodeBuilder.

getConstantFP()

IRValue * proteus::MLIRCodeBuilder::getConstantFP ( IRType  Ty, double  Val  )

overridevirtual

Implements proteus::CodeBuilder.

getConstantInt()

IRValue * proteus::MLIRCodeBuilder::getConstantInt ( IRType  Ty, uint64_t  Val  )

overridevirtual

Implements proteus::CodeBuilder.

getElementPtr() [1/2]

VarAlloc proteus::MLIRCodeBuilder::getElementPtr ( IRValue *  Base, IRType  BaseTy, IRValue *  Index, IRType  ElemTy  )

overridevirtual

Implements proteus::CodeBuilder.

getElementPtr() [2/2]

VarAlloc proteus::MLIRCodeBuilder::getElementPtr ( IRValue *  Base, IRType  BaseTy, size_t  Index, IRType  ElemTy  )

overridevirtual

Implements proteus::CodeBuilder.

getTargetModel()

TargetModelType proteus::MLIRCodeBuilder::getTargetModel ( ) const

inlineoverridevirtual

Implements proteus::CodeBuilder.

loadAddress()

IRValue * proteus::MLIRCodeBuilder::loadAddress ( IRValue *  Slot, IRType  AllocTy  )

overridevirtual

Load the pointer stored in Slot (pointer alloca).

Implements proteus::CodeBuilder.

loadFromPointee()

IRValue * proteus::MLIRCodeBuilder::loadFromPointee ( IRValue *  Slot, IRType  AllocTy, IRType  ValueTy  )

overridevirtual

Dereference the pointer stored in Slot, then load the pointee.

Implements proteus::CodeBuilder.

loadScalar()

IRValue * proteus::MLIRCodeBuilder::loadScalar ( IRValue *  Slot, IRType  ValueTy  )

overridevirtual

Load the value stored directly in Slot (scalar alloca).

Implements proteus::CodeBuilder.

operator=()

MLIRCodeBuilder & proteus::MLIRCodeBuilder::operator= ( const MLIRCodeBuilder &  )

delete

print()

void proteus::MLIRCodeBuilder::print

(

)

Dump the MLIR module to llvm::outs().

printLLVMIR()

void proteus::MLIRCodeBuilder::printLLVMIR

(

llvm::raw_ostream &

OS

)

Print the lowered LLVM IR. Useful for debugging the MLIR->LLVM lowering.

setDeviceArch()

void proteus::MLIRCodeBuilder::setDeviceArch

(

const std::string &

Arch

)

Device-only lowering configuration. For HIP this is the target chipset (e.g. gfx90a) used by ROCDL lowering. For CUDA it is currently unused.

setFunctionName()

void proteus::MLIRCodeBuilder::setFunctionName ( IRFunction *  F, const std::string &  Name  )

overridevirtual

Rename the function identified by F.

Implements proteus::CodeBuilder.

setInsertPointAtEntry()

void proteus::MLIRCodeBuilder::setInsertPointAtEntry ( )

overridevirtual

Implements proteus::CodeBuilder.

storeAddress()

void proteus::MLIRCodeBuilder::storeAddress ( IRValue *  Slot, IRValue *  Addr  )

overridevirtual

Store Addr into Slot (pointer alloca).

Implements proteus::CodeBuilder.

storeScalar()

void proteus::MLIRCodeBuilder::storeScalar ( IRValue *  Slot, IRValue *  Val  )

overridevirtual

Store Val directly into Slot (scalar alloca).

Implements proteus::CodeBuilder.

storeToPointee()

void proteus::MLIRCodeBuilder::storeToPointee ( IRValue *  Slot, IRType  AllocTy, IRValue *  Val  )

overridevirtual

Dereference the pointer stored in Slot, then store Val to it.

Implements proteus::CodeBuilder.

takeContext()

std::unique_ptr< llvm::LLVMContext > proteus::MLIRCodeBuilder::takeContext

(

)

Transfers ownership of the lowered LLVM context/module. These trigger lowering if it has not yet been performed. Calling order is flexible: takeContext() and takeModule() are safe in either order.

takeModule()

std::unique_ptr< llvm::Module > proteus::MLIRCodeBuilder::takeModule

(

)

---

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

• include/proteus/Frontend/MLIRCodeBuilder.h
• src/runtime/Frontend/MLIRCodeBuilder.cpp