#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)

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)

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. More...

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.

Function Documentation

◆ mgpuEventCreate()

hipEvent_t mgpuEventCreate ( )

Definition at line 89 of file RocmRuntimeWrappers.cpp.

References HIP_REPORT_IF_ERROR.

◆ mgpuEventDestroy()

void mgpuEventDestroy ( hipEvent_t event )

Definition at line 95 of file RocmRuntimeWrappers.cpp.

References HIP_REPORT_IF_ERROR.

◆ mgpuEventRecord()

void mgpuEventRecord	(	hipEvent_t	event,
		hipStream_t	stream
	)

Definition at line 103 of file RocmRuntimeWrappers.cpp.

References HIP_REPORT_IF_ERROR.

◆ mgpuEventSynchronize()

void mgpuEventSynchronize ( hipEvent_t event )

Definition at line 99 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
	)