#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
using	UniqueZeCommandList
using	UniqueZeEvent
using	UniqueZeEventPool

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.

Referenced by DynamicEventPool::createNewPool(), getContext(), getDevice(), getDriver(), L0RTContextWrapper::L0RTContextWrapper(), loadModule(), mgpuEventRecord(), mgpuEventSynchronize(), mgpuLaunchKernel(), mgpuMemAlloc(), mgpuMemcpy(), mgpuMemFree(), mgpuMemset(), mgpuModuleGetFunction(), mgpuModuleUnload(), ZeCommandListDeleter::operator()(), ZeContextDeleter::operator()(), ZeEventDeleter::operator()(), ZeEventPoolDeleter::operator()(), DynamicEventPool::releaseEvent(), StreamWrapper::sync(), and DynamicEventPool::takeEvent().

Typedef Documentation

◆ UniqueZeCommandList

using UniqueZeCommandList

Initial value:

 
    std::unique_ptr<std::remove_pointer<ze_command_list_handle_t>::type,
                    ZeCommandListDeleter>

Definition at line 136 of file LevelZeroRuntimeWrappers.cpp.

◆ UniqueZeContext

using UniqueZeContext

Initial value:

 
    std::unique_ptr<std::remove_pointer<ze_context_handle_t>::type,
                    ZeContextDeleter>

Definition at line 133 of file LevelZeroRuntimeWrappers.cpp.

◆ UniqueZeEvent

using UniqueZeEvent

Initial value:

 
    std::unique_ptr<std::remove_pointer<ze_event_handle_t>::type,
                    ZeEventDeleter>

Definition at line 234 of file LevelZeroRuntimeWrappers.cpp.

◆ UniqueZeEventPool

using UniqueZeEventPool

Initial value:

 
    std::unique_ptr<std::remove_pointer<ze_event_pool_handle_t>::type,
                    ZeEventPoolDeleter>

Definition at line 237 of file LevelZeroRuntimeWrappers.cpp.

Function Documentation

◆ getContext()

ze_context_handle_t getContext ( ze_driver_handle_t driver )

static

Definition at line 105 of file LevelZeroRuntimeWrappers.cpp.

References L0_SAFE_CALL.

◆ getDevice()

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(), and L0RTContextWrapper::L0RTContextWrapper().

◆ getDriver()

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(), and L0RTContextWrapper::L0RTContextWrapper().

◆ getDynamicEventPool()

DynamicEventPool & getDynamicEventPool ( )

static

Definition at line 336 of file LevelZeroRuntimeWrappers.cpp.

References getRtContext().

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

◆ getRtContext()

L0RTContextWrapper & getRtContext ( )

static

Definition at line 331 of file LevelZeroRuntimeWrappers.cpp.

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

◆ loadModule()

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(), L0_SAFE_CALL, and result.

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.

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