#include "llvm/ADT/Twine.h"
#include "level_zero/ze_api.h"
#include <cassert>
#include <deque>
#include <exception>
#include <functional>
#include <iostream>
#include <limits>
#include <unordered_set>
#include <vector>

Classes
struct	ZeContextDeleter

struct	ZeCommandListDeleter

struct	L0RTContextWrapper

struct	ZeEventDeleter

struct	ZeEventPoolDeleter

struct	DynamicEventPool

struct	StreamWrapper

Macros
#define	L0_SAFE_CALL(call)

Typedefs
using	UniqueZeContext = std::unique_ptr< std::remove_pointer< ze_context_handle_t >::type, ZeContextDeleter >

using	UniqueZeCommandList = std::unique_ptr< std::remove_pointer< ze_command_list_handle_t >::type, ZeCommandListDeleter >

using	UniqueZeEvent = std::unique_ptr< std::remove_pointer< ze_event_handle_t >::type, ZeEventDeleter >

using	UniqueZeEventPool = std::unique_ptr< std::remove_pointer< ze_event_pool_handle_t >::type, ZeEventPoolDeleter >

Functions
static ze_driver_handle_t	getDriver (uint32_t idx=0)

static ze_device_handle_t	getDevice (const uint32_t driverIdx=0, const int32_t devIdx=0)

static ze_context_handle_t	getContext (ze_driver_handle_t driver)

static L0RTContextWrapper &	getRtContext ()

static DynamicEventPool &	getDynamicEventPool ()

static ze_module_handle_t	loadModule (const void *data, size_t dataSize)

StreamWrapper *	mgpuStreamCreate ()

void	mgpuStreamSynchronize (StreamWrapper *stream)

void	mgpuStreamDestroy (StreamWrapper *stream)

void	mgpuStreamWaitEvent (StreamWrapper *stream, ze_event_handle_t event)

ze_event_handle_t	mgpuEventCreate ()

void	mgpuEventDestroy (ze_event_handle_t event)

void	mgpuEventSynchronize (ze_event_handle_t event)

void	mgpuEventRecord (ze_event_handle_t event, StreamWrapper *stream)

void *	mgpuMemAlloc (uint64_t size, StreamWrapper *stream, bool isShared)

void	mgpuMemFree (void ptr, StreamWrapper stream)

void	mgpuMemcpy (void dst, void src, size_t sizeBytes, StreamWrapper *stream)

template<typename PATTERN_TYPE >
static void	mgpuMemset (void dst, PATTERN_TYPE value, size_t count, StreamWrapper stream)

void	mgpuMemset32 (void dst, unsigned int value, size_t count, StreamWrapper stream)

void	mgpuMemset16 (void dst, unsigned short value, size_t count, StreamWrapper stream)

ze_module_handle_t	mgpuModuleLoad (const void *data, size_t gpuBlobSize)

ze_kernel_handle_t	mgpuModuleGetFunction (ze_module_handle_t module, const char *name)

void	mgpuLaunchKernel (ze_kernel_handle_t kernel, size_t gridX, size_t gridY, size_t gridZ, size_t blockX, size_t blockY, size_t blockZ, size_t sharedMemBytes, StreamWrapper stream, void params, void *, size_t paramsCount)

void	mgpuModuleUnload (ze_module_handle_t module)

void	mgpuSetDefaultDevice (int32_t devIdx)

Macro Definition Documentation

◆ L0_SAFE_CALL

#define L0_SAFE_CALL ( call )

Value:

  {                                                                            \
    ze_result_t status = (call);                                               \
    if (status != ZE_RESULT_SUCCESS) {                                         \
      const char *errorString;                                                 \
      zeDriverGetLastErrorDescription(NULL, &errorString);                     \
      std::cerr << "L0 error " << status << ": " << errorString << std::endl;  \
      std::abort();                                                            \
    }                                                                          \
  }

Definition at line 39 of file LevelZeroRuntimeWrappers.cpp.

Typedef Documentation

◆ UniqueZeCommandList

using UniqueZeCommandList = std::unique_ptr<std::remove_pointer<ze_command_list_handle_t>::type, ZeCommandListDeleter>

Definition at line 136 of file LevelZeroRuntimeWrappers.cpp.

◆ UniqueZeContext

using UniqueZeContext = std::unique_ptr<std::remove_pointer<ze_context_handle_t>::type, ZeContextDeleter>

Definition at line 133 of file LevelZeroRuntimeWrappers.cpp.

◆ UniqueZeEvent

using UniqueZeEvent = std::unique_ptr<std::remove_pointer<ze_event_handle_t>::type, ZeEventDeleter>

Definition at line 234 of file LevelZeroRuntimeWrappers.cpp.

◆ UniqueZeEventPool

using UniqueZeEventPool = std::unique_ptr<std::remove_pointer<ze_event_pool_handle_t>::type, ZeEventPoolDeleter>

Definition at line 237 of file LevelZeroRuntimeWrappers.cpp.

Function Documentation

◆ getContext()

static ze_context_handle_t getContext ( ze_driver_handle_t driver )

static

Definition at line 105 of file LevelZeroRuntimeWrappers.cpp.

References L0_SAFE_CALL.

Referenced by L0RTContextWrapper::L0RTContextWrapper().

◆ getDevice()

static ze_device_handle_t getDevice	(	const uint32_t	driverIdx = `0`,
		const int32_t	devIdx = `0`
	)

static

Definition at line 82 of file LevelZeroRuntimeWrappers.cpp.

References getDriver(), and L0_SAFE_CALL.

Referenced by getTargetEntryUniqueInfo().

◆ getDriver()

static ze_driver_handle_t getDriver ( uint32_t idx = 0 )

static

Definition at line 58 of file LevelZeroRuntimeWrappers.cpp.

References L0_SAFE_CALL.

Referenced by getDevice().

◆ getDynamicEventPool()

static DynamicEventPool& getDynamicEventPool ( )

static

Definition at line 336 of file LevelZeroRuntimeWrappers.cpp.

References getRtContext().

Referenced by mgpuEventCreate(), mgpuEventDestroy(), mgpuSetDefaultDevice(), and mgpuStreamCreate().

◆ getRtContext()

static L0RTContextWrapper& getRtContext ( )

static

Definition at line 331 of file LevelZeroRuntimeWrappers.cpp.

Referenced by getDynamicEventPool(), loadModule(), mgpuEventRecord(), mgpuLaunchKernel(), mgpuMemAlloc(), mgpuMemcpy(), mgpuMemFree(), mgpuMemset(), and mgpuSetDefaultDevice().

◆ loadModule()

static ze_module_handle_t loadModule	(	const void *	data,
		size_t	dataSize
	)

static

Definition at line 383 of file LevelZeroRuntimeWrappers.cpp.

References L0RTContextWrapper::context, L0RTContextWrapper::device, getRtContext(), and L0_SAFE_CALL.

Referenced by mgpuModuleLoad().

◆ mgpuEventCreate()

ze_event_handle_t mgpuEventCreate ( )

Definition at line 432 of file LevelZeroRuntimeWrappers.cpp.

References getDynamicEventPool(), and DynamicEventPool::takeEvent().

◆ mgpuEventDestroy()

void mgpuEventDestroy ( ze_event_handle_t event )

Definition at line 436 of file LevelZeroRuntimeWrappers.cpp.

References getDynamicEventPool(), and DynamicEventPool::releaseEvent().

◆ mgpuEventRecord()

void mgpuEventRecord	(	ze_event_handle_t	event,
		StreamWrapper *	stream
	)

Definition at line 446 of file LevelZeroRuntimeWrappers.cpp.

References getRtContext(), L0RTContextWrapper::immCmdListCompute, L0RTContextWrapper::immCmdListCopy, and L0_SAFE_CALL.

◆ mgpuEventSynchronize()

void mgpuEventSynchronize ( ze_event_handle_t event )

Definition at line 440 of file LevelZeroRuntimeWrappers.cpp.

References L0_SAFE_CALL, and max().

◆ mgpuLaunchKernel()

void mgpuLaunchKernel	(	ze_kernel_handle_t	kernel,
		size_t	gridX,
		size_t	gridY,
		size_t	gridZ,
		size_t	blockX,
		size_t	blockY,
		size_t	blockZ,
		size_t	sharedMemBytes,
		StreamWrapper *	stream,
		void **	params,
		void **	,
		size_t	paramsCount
	)