MLIR: lib/Dialect/GPU/TransformOps/Utils.cpp File Reference (original) (raw)
Go to the source code of this file.
| Namespaces |
|---|
| Macros | |
|---|---|
| #define | DEBUG_TYPE "gpu-transforms" |
| #define | DBGS() (llvm::dbgs() << '[' << DEBUG_TYPE << "] ") |
| #define | LDBG(X) LLVM_DEBUG(DBGS() << (X) << "\n") |
| #define | DBGS_ALIAS() (llvm::dbgs() << '[' << DEBUG_TYPE_ALIAS << "] ") |
| Functions | |
|---|---|
| template | |
| static Value | buildLinearId (RewriterBase &rewriter, Location loc, ArrayRef< OpFoldResult > originalBasisOfr) |
| Return a flattened thread id for the workgroup with given sizes. More... | |
| template | |
| static GpuIdBuilderFnType | commonLinearIdBuilderFn (int64_t multiplicity=1) |
| Create a linear id builder that takes the originalBasisOfr and decompose it in the basis of forallMappingSizes. More... | |
| template | |
| static GpuIdBuilderFnType | common3DIdBuilderFn (int64_t multiplicity=1) |
| Create a simple 3-D id builder that takes the originalBasisOfr The 3-D id builder returns a 3-D vector of ids for indexing and 3-D sizes. More... | |
| DiagnosedSilenceableFailure | mlir::transform::gpu::checkGpuLimits (TransformOpInterface transformOp, std::optional< int64_t > gridDimX, std::optional< int64_t > gridDimY, std::optional< int64_t > gridDimZ, std::optional< int64_t > blockDimX, std::optional< int64_t > blockDimY, std::optional< int64_t > blockDimZ) |
| Determine if the size of the kernel configuration is supported by the GPU architecture being used. More... | |
| DiagnosedSilenceableFailure | mlir::transform::gpu::createGpuLaunch (RewriterBase &rewriter, Location loc, TransformOpInterface transformOp, LaunchOp &launchOp, std::optional< int64_t > gridDimX, std::optional< int64_t > gridDimY, std::optional< int64_t > gridDimZ, std::optional< int64_t > blockDimX, std::optional< int64_t > blockDimY, std::optional< int64_t > blockDimZ) |
| DiagnosedSilenceableFailure | mlir::transform::gpu::alterGpuLaunch (RewriterBase &rewriter, LaunchOp gpuLaunch, TransformOpInterface transformOp, std::optional< int64_t > gridDimX, std::optional< int64_t > gridDimY, std::optional< int64_t > gridDimZ, std::optional< int64_t > blockDimX, std::optional< int64_t > blockDimY, std::optional< int64_t > blockDimZ) |
| Alter kernel configuration of the given kernel. More... | |
◆ DBGS
| #define DBGS | ( | ) | (llvm::dbgs() << '[' << DEBUG_TYPE << "] ") |
|---|
◆ DBGS_ALIAS
◆ DEBUG_TYPE
#define DEBUG_TYPE "gpu-transforms"
◆ LDBG
| #define LDBG | ( | | X | ) | LLVM_DEBUG(DBGS() << (X) << "\n") | | ------------ | - | | - | - | ------------------------------------------------------------------------------------------------ |
◆ buildLinearId()
template
◆ common3DIdBuilderFn()
template
◆ commonLinearIdBuilderFn()
template