File elemwise.h¶
Custom elementwise operations generator.
Typedefs
-
typedef struct _GpuElemwise GpuElemwise¶
Elementwise generator structure.
The contents are private.
Functions
-
GpuElemwise *GpuElemwise_new(gpucontext *ctx, const char *preamble, const char *expr, unsigned int n, gpuelemwise_arg *args, unsigned int nd, int flags)¶
Create a new GpuElemwise.
This will allocate and initialized a new GpuElemwise object. This object can be used to run the specified operation on different sets of arrays.
The argument descriptor name the arguments and provide their data types and geometry (arrays or scalars). They also specify if the arguments are used for reading or writing. An argument can be used for both.
The expression is a C-like string performing an operation with scalar values named according to the argument descriptors. All of the indexing and selection of the right values is handled by the GpuElemwise code.
- Parameters:
ctx – the context in which to run the operations
preamble – code to be inserted before the kernel code
expr – the expression to compute
n – the number of arguments
args – the argument descriptors
nd – the number of dimensions to precompile for
flags – see GpuElemwise flags
- Returns:
a new GpuElemwise object or NULL
-
void GpuElemwise_free(GpuElemwise *ge)¶
Free all storage associated with a GpuElemwise.
- Parameters:
ge – the GpuElemwise object to free.
-
int GpuElemwise_call(GpuElemwise *ge, void **args, int flags)¶
Run a GpuElemwise on some inputs.
- Parameters:
ge – the GpuElemwise to run
args – pointers to the arguments (must macth what was described by the argument descriptors)
flags – see GpuElemwise call flags
-
struct gpuelemwise_arg¶
- #include <elemwise.h>
Argument information structure for GpuElemwise.