MLIR  22.0.0git
Classes | Namespaces | Enumerations | Functions
BasicPtxBuilderInterface.h File Reference
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
#include "mlir/IR/BuiltinAttributes.h"
#include "mlir/IR/PatternMatch.h"
#include "mlir/IR/Value.h"
#include "llvm/Support/LogicalResult.h"
#include "mlir/Dialect/LLVMIR/BasicPtxBuilderInterface.h.inc"

Go to the source code of this file.

Classes

class  mlir::NVVM::PtxBuilder
 A class to build PTX assembly automatically. More...
 

Namespaces

 mlir
 Include the generated interface declarations.
 
 mlir::NVVM
 

Enumerations

enum class  mlir::NVVM::PTXRegisterMod { mlir::NVVM::Read = 0 , mlir::NVVM::Write = 2 , mlir::NVVM::ReadWrite = 1 }
 Register read/write modifier to build constraint string for PTX inline https://docs.nvidia.com/cuda/inline-ptx-assembly/index.html#parameters. More...
 

Functions

llvm::raw_ostream & mlir::NVVM::operator<< (llvm::raw_ostream &os, PTXRegisterMod mod)
 
void mlir::NVVM::countPlaceholderNumbers (StringRef ptxCode, llvm::SmallDenseSet< unsigned > &seenRW, llvm::SmallDenseSet< unsigned > &seenW, llvm::SmallDenseSet< unsigned > &seenR, llvm::SmallVectorImpl< unsigned > &rwNums, llvm::SmallVectorImpl< unsigned > &wNums, llvm::SmallVectorImpl< unsigned > &rNums)
 Count the number of placeholder variables such as {$r}, {$w}, {$rw} in the PTX code. More...