rocfft_execution_info_set_load_callback Interface Reference

rocfft_execution_info_set_load_callback Interface Reference#

HIPFORT API Reference: hipfort_rocfft::rocfft_execution_info_set_load_callback Interface Reference

Set a load callback for a plan execution (experimental) More...

Public Member Functions
integer(kind(rocfft_status_success)) function	rocfft_execution_info_set_load_callback_ (myInfo, cb_functions, cb_data, shared_mem_bytes)

Detailed Description

Set a load callback for a plan execution (experimental)

This function specifies a user-defined callback function that is run to load input from global memory at the start of the transform. Callbacks are an experimental feature in rocFFT.

Callback function pointers/data are given as arrays, with one function/data pointer per device executing this plan. Currently, plans can only use one device.

The provided function pointers replace any previously-specified load callback for this execution info handle.

Load callbacks have the following signature:

t load_cb(t* data, size_t offset, void* cbdata, void* sharedmem);

'T' is the type of a single element of the input buffer. It is the caller's responsibility to ensure that the function type is appropriate for the plan (for example, a single-precision real-to-complex transform would load single-precision real elements).

A null value for 'cb' may be specified to clear any previously registered load callback.

Currently, 'shared_mem_bytes' must be 0. Callbacks are not supported on transforms that use planar formats for either input or output.

Parameters

[in]	info	execution info handle
[in]	cb	callback function pointers
[in]	cbdata	callback function data, passed to the function pointer when it is called
[in]	shared_mem_bytes	amount of shared memory to allocate for the callback function to use

Member Function/Subroutine Documentation