Peano
Loading...
Searching...
No Matches
MLIRbridgeHIPfunctions.h File Reference
#include <iostream>
#include <cstring>
#include <cstdint>
#include <cmath>
#include <cstdio>
#include <iomanip>
#include <cstdlib>
Include dependency graph for MLIRbridgeHIPfunctions.h:

Go to the source code of this file.

Typedefs

typedef void * GPUModule
 
typedef void * GPUFunction
 
typedef void * GPUStream
 

Functions

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)
 

Typedef Documentation

◆ GPUFunction

typedef void* GPUFunction

Definition at line 16 of file MLIRbridgeHIPfunctions.h.

◆ GPUModule

typedef void* GPUModule

Definition at line 15 of file MLIRbridgeHIPfunctions.h.

◆ GPUStream

typedef void* GPUStream

Definition at line 17 of file MLIRbridgeHIPfunctions.h.

Function Documentation

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

GPUFunction mgpuModuleGetFunction ( GPUModule  module,
const char *  name 
)

◆ mgpuModuleLoad()

GPUModule mgpuModuleLoad ( const void *  binary,
size_t  size 
)

◆ mgpuModuleUnload()

void mgpuModuleUnload ( GPUModule  module)

◆ mgpuStreamCreate()

GPUStream mgpuStreamCreate ( )

◆ mgpuStreamDestroy()

void mgpuStreamDestroy ( GPUStream  stream)

◆ mgpuStreamSynchronize()

void mgpuStreamSynchronize ( GPUStream  stream)

◆ releaseMLIRContextGPU()

void releaseMLIRContextGPU ( hipDevice_t  targetDevice)

◆ setupMLIRContextGPU()

void setupMLIRContextGPU ( hipDevice_t  targetDevice)