MLIR
22.0.0git
|
#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>
Go to the source code of this file.
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) |
L0RTContextWrapper & | getRtContext () |
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 > | |
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) |
#define L0_SAFE_CALL | ( | call | ) |
Definition at line 39 of file LevelZeroRuntimeWrappers.cpp.
using UniqueZeCommandList = std::unique_ptr<std::remove_pointer<ze_command_list_handle_t>::type, ZeCommandListDeleter> |
Definition at line 136 of file LevelZeroRuntimeWrappers.cpp.
using UniqueZeContext = std::unique_ptr<std::remove_pointer<ze_context_handle_t>::type, ZeContextDeleter> |
Definition at line 133 of file LevelZeroRuntimeWrappers.cpp.
using UniqueZeEvent = std::unique_ptr<std::remove_pointer<ze_event_handle_t>::type, ZeEventDeleter> |
Definition at line 234 of file LevelZeroRuntimeWrappers.cpp.
using UniqueZeEventPool = std::unique_ptr<std::remove_pointer<ze_event_pool_handle_t>::type, ZeEventPoolDeleter> |
Definition at line 237 of file LevelZeroRuntimeWrappers.cpp.
|
static |
Definition at line 105 of file LevelZeroRuntimeWrappers.cpp.
References L0_SAFE_CALL.
Referenced by L0RTContextWrapper::L0RTContextWrapper().
|
static |
Definition at line 82 of file LevelZeroRuntimeWrappers.cpp.
References getDriver(), and L0_SAFE_CALL.
Referenced by getTargetEntryUniqueInfo().
|
static |
Definition at line 58 of file LevelZeroRuntimeWrappers.cpp.
References L0_SAFE_CALL.
Referenced by getDevice().
DynamicEventPool& getDynamicEventPool | ( | ) |
Definition at line 336 of file LevelZeroRuntimeWrappers.cpp.
References getRtContext().
Referenced by mgpuEventCreate(), mgpuEventDestroy(), mgpuSetDefaultDevice(), and mgpuStreamCreate().
L0RTContextWrapper& getRtContext | ( | ) |
Definition at line 331 of file LevelZeroRuntimeWrappers.cpp.
Referenced by getDynamicEventPool(), loadModule(), mgpuEventRecord(), mgpuLaunchKernel(), mgpuMemAlloc(), mgpuMemcpy(), mgpuMemFree(), mgpuMemset(), and mgpuSetDefaultDevice().
|
static |
Definition at line 383 of file LevelZeroRuntimeWrappers.cpp.
References L0RTContextWrapper::context, L0RTContextWrapper::device, getRtContext(), and L0_SAFE_CALL.
Referenced by mgpuModuleLoad().
ze_event_handle_t mgpuEventCreate | ( | ) |
Definition at line 432 of file LevelZeroRuntimeWrappers.cpp.
References getDynamicEventPool(), and DynamicEventPool::takeEvent().
void mgpuEventDestroy | ( | ze_event_handle_t | event | ) |
Definition at line 436 of file LevelZeroRuntimeWrappers.cpp.
References getDynamicEventPool(), and DynamicEventPool::releaseEvent().
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.
void mgpuEventSynchronize | ( | ze_event_handle_t | event | ) |
Definition at line 440 of file LevelZeroRuntimeWrappers.cpp.
References L0_SAFE_CALL, and max().
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 | ||
) |
Definition at line 534 of file LevelZeroRuntimeWrappers.cpp.
References StreamWrapper::enqueueOp(), getRtContext(), L0RTContextWrapper::immCmdListCompute, and L0_SAFE_CALL.
void* mgpuMemAlloc | ( | uint64_t | size, |
StreamWrapper * | stream, | ||
bool | isShared | ||
) |
Definition at line 454 of file LevelZeroRuntimeWrappers.cpp.
References L0RTContextWrapper::context, L0RTContextWrapper::device, getRtContext(), and L0_SAFE_CALL.
void mgpuMemcpy | ( | void * | dst, |
void * | src, | ||
size_t | sizeBytes, | ||
StreamWrapper * | stream | ||
) |
Definition at line 484 of file LevelZeroRuntimeWrappers.cpp.
References StreamWrapper::enqueueOp(), getRtContext(), L0RTContextWrapper::immCmdListCopy, and L0_SAFE_CALL.
void mgpuMemFree | ( | void * | ptr, |
StreamWrapper * | stream | ||
) |
Definition at line 478 of file LevelZeroRuntimeWrappers.cpp.
References L0RTContextWrapper::context, getRtContext(), L0_SAFE_CALL, and StreamWrapper::sync().
void mgpuMemset | ( | void * | dst, |
PATTERN_TYPE | value, | ||
size_t | count, | ||
StreamWrapper * | stream | ||
) |
Definition at line 495 of file LevelZeroRuntimeWrappers.cpp.
References L0RTContextWrapper::copyEngineMaxMemoryFillPatternSize, StreamWrapper::enqueueOp(), getRtContext(), L0RTContextWrapper::immCmdListCompute, L0RTContextWrapper::immCmdListCopy, and L0_SAFE_CALL.
void mgpuMemset16 | ( | void * | dst, |
unsigned short | value, | ||
size_t | count, | ||
StreamWrapper * | stream | ||
) |
Definition at line 514 of file LevelZeroRuntimeWrappers.cpp.
void mgpuMemset32 | ( | void * | dst, |
unsigned int | value, | ||
size_t | count, | ||
StreamWrapper * | stream | ||
) |
Definition at line 509 of file LevelZeroRuntimeWrappers.cpp.
ze_kernel_handle_t mgpuModuleGetFunction | ( | ze_module_handle_t | module, |
const char * | name | ||
) |
Definition at line 524 of file LevelZeroRuntimeWrappers.cpp.
References L0_SAFE_CALL.
ze_module_handle_t mgpuModuleLoad | ( | const void * | data, |
size_t | gpuBlobSize | ||
) |
Definition at line 519 of file LevelZeroRuntimeWrappers.cpp.
References loadModule().
void mgpuModuleUnload | ( | ze_module_handle_t | module | ) |
Definition at line 562 of file LevelZeroRuntimeWrappers.cpp.
References L0_SAFE_CALL.
void mgpuSetDefaultDevice | ( | int32_t | devIdx | ) |
Definition at line 566 of file LevelZeroRuntimeWrappers.cpp.
References getDynamicEventPool(), getRtContext(), and L0RTContextWrapper::L0RTContextWrapper().
StreamWrapper* mgpuStreamCreate | ( | ) |
Definition at line 414 of file LevelZeroRuntimeWrappers.cpp.
References getDynamicEventPool().
void mgpuStreamDestroy | ( | StreamWrapper * | stream | ) |
Definition at line 423 of file LevelZeroRuntimeWrappers.cpp.
void mgpuStreamSynchronize | ( | StreamWrapper * | stream | ) |
Definition at line 418 of file LevelZeroRuntimeWrappers.cpp.
References StreamWrapper::sync().
void mgpuStreamWaitEvent | ( | StreamWrapper * | stream, |
ze_event_handle_t | event | ||
) |
Definition at line 425 of file LevelZeroRuntimeWrappers.cpp.
References StreamWrapper::sync().