MLIR
20.0.0git
|
This is the complete list of members for CopyBuilder, including all inherited members.
buildAndInitBarrierInSharedMemory(OpFoldResult numThreads) | HopperBuilder | |
buildBarrierArriveTx(TypedValue< nvgpu::MBarrierGroupType > barrier, ArrayRef< OpFoldResult > sizes) | HopperBuilder | |
buildGlobalMemRefDescriptor(TypedValue< MemRefType > memref, gpu::LaunchOp launchOp) | HopperBuilder | |
buildPredicateLoadsOnThread0(ArrayRef< TypedValue< nvgpu::TensorMapDescriptorType >> globalDescriptors, ArrayRef< TypedValue< MemRefType >> sharedMemBuffers, TypedValue< nvgpu::MBarrierGroupType > barrier) | HopperBuilder | |
buildTmaAsyncLoad(TypedValue< nvgpu::TensorMapDescriptorType > globalDesc, TypedValue< MemRefType > sharedMemref, TypedValue< nvgpu::MBarrierGroupType > barrier, SmallVectorImpl< Operation * > &loadOps) | HopperBuilder | |
buildTryWaitParity(TypedValue< nvgpu::MBarrierGroupType > barrier) | HopperBuilder | |
CopyBuilder(RewriterBase &rewriter, Location loc) | CopyBuilder | inline |
HopperBuilder(RewriterBase &rewriter, Location loc) | HopperBuilder | inline |
loc | HopperBuilder | |
rewrite(ArrayRef< Operation * > copyOps) | CopyBuilder | |
rewriter | HopperBuilder |