tvm_tirx.backend.trn.transform.LowerTIRx

Contents

tvm_tirx.backend.trn.transform.LowerTIRx#

tvm_tirx.backend.trn.transform.LowerTIRx()#

Run TileDispatch followed by LowerTrainiumLayout.

Run TrnPrivateTensorAlloc and TrnNaiveAllocator first when tiles need workspaces or allocation addresses. Tile dispatch requires a Trainium target on the function or in the current target scope.