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 | |
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... | |