|
MLIR 22.0.0git
|
#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 | |
| namespace | mlir |
| Include the generated interface declarations. | |
| namespace | 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. | |