#include <iostream>
#include <cstring>
#include <cstdint>
#include <cmath>
#include <cstdio>
#include <iomanip>
#include <cstdlib>
Go to the source code of this file.
|
| GPUModule | mgpuModuleLoad (const void *binary, size_t size) |
| |
| void | mgpuModuleUnload (GPUModule module) |
| |
| GPUFunction | mgpuModuleGetFunction (GPUModule module, const char *name) |
| |
| GPUStream | mgpuStreamCreate () |
| |
| void | mgpuStreamDestroy (GPUStream stream) |
| |
| void | mgpuStreamSynchronize (GPUStream stream) |
| |
| void | mgpuLaunchKernel (GPUFunction func, int64_t gridDimX, int64_t gridDimY, int64_t gridDimZ, int64_t blockDimX, int64_t blockDimY, int64_t blockDimZ, int32_t smemSize, GPUStream stream, void **kernelParams, void **extraParams, int64_t paramCount) |
| |
| void * | mgpuMemAlloc (int64_t size, void *stream, int8_t flags) |
| |
| int32_t | mgpuMemFree (void *ptr, void *stream) |
| |
| void | setupMLIRContextGPU (hipDevice_t targetDevice) |
| |
| void | releaseMLIRContextGPU (hipDevice_t targetDevice) |
| |
◆ GPUFunction
◆ GPUModule
◆ GPUStream
◆ mgpuLaunchKernel()
| void mgpuLaunchKernel |
( |
GPUFunction |
func, |
|
|
int64_t |
gridDimX, |
|
|
int64_t |
gridDimY, |
|
|
int64_t |
gridDimZ, |
|
|
int64_t |
blockDimX, |
|
|
int64_t |
blockDimY, |
|
|
int64_t |
blockDimZ, |
|
|
int32_t |
smemSize, |
|
|
GPUStream |
stream, |
|
|
void ** |
kernelParams, |
|
|
void ** |
extraParams, |
|
|
int64_t |
paramCount |
|
) |
| |
◆ mgpuMemAlloc()
| void * mgpuMemAlloc |
( |
int64_t |
size, |
|
|
void * |
stream, |
|
|
int8_t |
flags |
|
) |
| |
◆ mgpuMemFree()
| int32_t mgpuMemFree |
( |
void * |
ptr, |
|
|
void * |
stream |
|
) |
| |
◆ mgpuModuleGetFunction()
◆ mgpuModuleLoad()
| GPUModule mgpuModuleLoad |
( |
const void * |
binary, |
|
|
size_t |
size |
|
) |
| |
◆ mgpuModuleUnload()
◆ mgpuStreamCreate()
◆ mgpuStreamDestroy()
◆ mgpuStreamSynchronize()
| void mgpuStreamSynchronize |
( |
GPUStream |
stream | ) |
|
◆ releaseMLIRContextGPU()
| void releaseMLIRContextGPU |
( |
hipDevice_t |
targetDevice | ) |
|
◆ setupMLIRContextGPU()
| void setupMLIRContextGPU |
( |
hipDevice_t |
targetDevice | ) |
|