#include <cassert>
#include <numeric>
#include "mlir/ExecutionEngine/CRunnerUtils.h"
#include "llvm/ADT/ArrayRef.h"
#include "hip/hip_runtime.h"

Macros
#define	HIP_REPORT_IF_ERROR(expr)

Functions
hipModule_t	mgpuModuleLoad (void *data, size_t)
hipModule_t	mgpuModuleLoadJIT (void *data, int optLevel, size_t)
void	mgpuModuleUnload (hipModule_t module)
hipFunction_t	mgpuModuleGetFunction (hipModule_t module, const char *name)
void	mgpuLaunchKernel (hipFunction_t function, intptr_t gridX, intptr_t gridY, intptr_t gridZ, intptr_t blockX, intptr_t blockY, intptr_t blockZ, int32_t smem, hipStream_t stream, void params, void extra, size_t)
void	mgpuLaunchKernelCooperative (hipFunction_t function, intptr_t gridX, intptr_t gridY, intptr_t gridZ, intptr_t clusterX, intptr_t clusterY, intptr_t clusterZ, intptr_t blockX, intptr_t blockY, intptr_t blockZ, int32_t smem, hipStream_t stream, void params, void )
hipStream_t	mgpuStreamCreate ()
void	mgpuStreamDestroy (hipStream_t stream)
void	mgpuStreamSynchronize (hipStream_t stream)
void	mgpuStreamWaitEvent (hipStream_t stream, hipEvent_t event)
hipEvent_t	mgpuEventCreate ()
void	mgpuEventDestroy (hipEvent_t event)
void	mgpuEventSynchronize (hipEvent_t event)
void	mgpuEventRecord (hipEvent_t event, hipStream_t stream)
void *	mgpuMemAlloc (uint64_t sizeBytes, hipStream_t, bool)
void	mgpuMemFree (void *ptr, hipStream_t)
void	mgpuMemcpy (void dst, void src, size_t sizeBytes, hipStream_t stream)
void	mgpuMemset32 (void *dst, int value, size_t count, hipStream_t stream)
void	mgpuMemset16 (void *dst, int short value, size_t count, hipStream_t stream)
void	mgpuMemHostRegister (void *ptr, uint64_t sizeBytes)
	Helper functions for writing mlir example code.
void	mgpuMemHostRegisterMemRef (int64_t rank, StridedMemRefType< char, 1 > *descriptor, int64_t elementSizeBytes)
void	mgpuMemHostUnregister (void *ptr)
void	mgpuMemHostUnregisterMemRef (int64_t rank, StridedMemRefType< char, 1 > *descriptor, int64_t elementSizeBytes)
template<typename T>
void	mgpuMemGetDevicePointer (T hostPtr, T *devicePtr)
StridedMemRefType< float, 1 >	mgpuMemGetDeviceMemRef1dFloat (float allocated, float aligned, int64_t offset, int64_t size, int64_t stride)
StridedMemRefType< int32_t, 1 >	mgpuMemGetDeviceMemRef1dInt32 (int32_t allocated, int32_t aligned, int64_t offset, int64_t size, int64_t stride)
void	mgpuSetDefaultDevice (int32_t device)

Variables
static thread_local int32_t	defaultDevice = 0

Macro Definition Documentation

◆ HIP_REPORT_IF_ERROR

#define HIP_REPORT_IF_ERROR ( expr )

Value:

  [](hipError_t result) {                                                      \
    if (!result)                                                               \
      return;                                                                  \
    const char *name = hipGetErrorName(result);                                \
    if (!name)                                                                 \
      name = "<unknown>";                                                      \
    fprintf(stderr, "'%s' failed with '%s'\n", #expr, name);                   \
  }(expr)

Definition at line 23 of file RocmRuntimeWrappers.cpp.

Referenced by mgpuEventCreate(), mgpuEventDestroy(), mgpuEventRecord(), mgpuEventSynchronize(), mgpuLaunchKernel(), mgpuLaunchKernelCooperative(), mgpuMemAlloc(), mgpuMemcpy(), mgpuMemFree(), mgpuMemGetDevicePointer(), mgpuMemHostRegister(), mgpuMemHostUnregister(), mgpuMemset16(), mgpuMemset32(), mgpuModuleGetFunction(), mgpuModuleLoad(), mgpuModuleUnload(), mgpuSetDefaultDevice(), mgpuStreamCreate(), mgpuStreamDestroy(), mgpuStreamSynchronize(), and mgpuStreamWaitEvent().

Function Documentation

◆ mgpuEventCreate()

hipEvent_t mgpuEventCreate ( )

Definition at line 110 of file RocmRuntimeWrappers.cpp.

References HIP_REPORT_IF_ERROR.

◆ mgpuEventDestroy()

void mgpuEventDestroy ( hipEvent_t event )

Definition at line 116 of file RocmRuntimeWrappers.cpp.

References HIP_REPORT_IF_ERROR.

◆ mgpuEventRecord()

void mgpuEventRecord	(	hipEvent_t	event,
		hipStream_t	stream )

Definition at line 124 of file RocmRuntimeWrappers.cpp.

References HIP_REPORT_IF_ERROR.

◆ mgpuEventSynchronize()

void mgpuEventSynchronize ( hipEvent_t event )

Definition at line 120 of file RocmRuntimeWrappers.cpp.

References HIP_REPORT_IF_ERROR.

◆ mgpuLaunchKernel()

void mgpuLaunchKernel	(	hipFunction_t	function,
		intptr_t	gridX,
		intptr_t	gridY,
		intptr_t	gridZ,
		intptr_t	blockX,
		intptr_t	blockY,
		intptr_t	blockZ,
		int32_t	smem,
		hipStream_t	stream,
		void **	params,
		void **	extra,
		size_t	)

Definition at line 61 of file RocmRuntimeWrappers.cpp.

References HIP_REPORT_IF_ERROR.

◆ mgpuLaunchKernelCooperative()

void mgpuLaunchKernelCooperative	(	hipFunction_t	function,
		intptr_t	gridX,
		intptr_t	gridY,
		intptr_t	gridZ,
		intptr_t	clusterX,
		intptr_t	clusterY,
		intptr_t	clusterZ,
		intptr_t	blockX,
		intptr_t	blockY,
		intptr_t	blockZ,
		int32_t	smem,
		hipStream_t	stream,
		void **	params,
		void **	)