mlir::ROCDL Namespace Reference

Classes
class	SerializeGPUModuleBase
	Base class for all ROCDL serializations from GPU modules into binary strings. 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...

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 87 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 72 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 79 of file Target.cpp.

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