#include "mlir/Dialect/GPU/IR/GPUDialect.h"
#include "mlir/IR/BuiltinOps.h"
#include "mlir/IR/Value.h"
#include "llvm/ADT/DenseMap.h"
#include <optional>
Go to the source code of this file.
|
| namespace | mlir |
| | Include the generated interface declarations.
|
| namespace | mlir::acc |
|
| std::optional< gpu::GPUModuleOp > | mlir::acc::getOrCreateGPUModule (ModuleOp mod, bool create=true, llvm::StringRef name=kDefaultGPUModuleName) |
| | Get or create a GPU module in the given module.
|
| Value | mlir::acc::getGPUSize (gpu::Processor processor, gpu::LaunchOp launch, const llvm::DenseMap< gpu::Processor, Value > &dimensionOps) |
| | Return the launch dimension for processor from launch, or from dimensionOps when launch is null.
|
| Value | mlir::acc::getGPUThreadId (gpu::Processor processor, gpu::LaunchOp launch, const llvm::DenseMap< gpu::Processor, Value > &indexOps) |
| | Return the thread/block index for processor from launch, or from indexOps when launch is null.
|