mlir::ROCDL Namespace Reference

Classes
class	SerializeGPUModuleBase
	Base class for all ROCDL serializations from GPU modules into binary strings. More...

Enumerations
enum class	AMDGCNLibraries : uint32_t {   None = 0 , Ockl = 1 , Ocml = 2 , OpenCL = 4 ,   Hip = 8 , LastLib = Hip , LLVM_MARK_AS_BITMASK_ENUM =(LastLib) , All = (LastLib << 1) - 1 }
	Helper enum for specifying the AMD GCN device libraries required for compilation. More...

Functions
void	registerROCDLTargetInterfaceExternalModels (DialectRegistry &registry)
	Registers the TargetAttrInterface for the #rocdl.target attribute in the given registry. More...
void	registerROCDLTargetInterfaceExternalModels (MLIRContext &context)
	Registers the TargetAttrInterface for the #rocdl.target attribute in the registry associated with the given context. More...
StringRef	getROCMPath ()
	Searches & returns the path ROCM toolkit path, the search order is: More...

Enumeration Type Documentation

AMDGCNLibraries

enum mlir::ROCDL::AMDGCNLibraries : uint32_t

strong

Helper enum for specifying the AMD GCN device libraries required for compilation.

Enumerator
None
Ockl
Ocml
OpenCL
Hip
LastLib
LLVM_MARK_AS_BITMASK_ENUM
All

Definition at line 33 of file Utils.h.

Function Documentation

getROCMPath()

StringRef mlir::ROCDL::getROCMPath

(

)

Searches & returns the path ROCM toolkit path, the search order is:

1. The ROCM_PATH environment variable.
2. The ROCM_ROOT environment variable.
3. The ROCM_HOME environment variable.
4. The ROCM path detected by CMake.
5. Returns an empty string.

Definition at line 84 of file Target.cpp.

References __DEFAULT_ROCM_PATH__.

Referenced by mlir::ROCDL::SerializeGPUModuleBase::SerializeGPUModuleBase().

registerROCDLTargetInterfaceExternalModels() [1/2]

void mlir::ROCDL::registerROCDLTargetInterfaceExternalModels

(

DialectRegistry &

registry

)

Registers the TargetAttrInterface for the #rocdl.target attribute in the given registry.

Definition at line 69 of file Target.cpp.

References mlir::DialectRegistry::addExtension().

Referenced by mlir::registerAllDialects(), and registerROCDLTargetInterfaceExternalModels().

registerROCDLTargetInterfaceExternalModels() [2/2]

void mlir::ROCDL::registerROCDLTargetInterfaceExternalModels

(

MLIRContext &

context

)

Registers the TargetAttrInterface for the #rocdl.target attribute in the registry associated with the given context.

Definition at line 76 of file Target.cpp.

References mlir::MLIRContext::appendDialectRegistry(), and registerROCDLTargetInterfaceExternalModels().