aclrtAllocatorSetAllocFuncToDesc

Product	Supported
Atlas 350 Accelerator Card	√
Atlas A3 training product/Atlas A3 inference product	√
Atlas A2 training product/Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	√
Atlas inference product	√
Atlas training product	√

Sets the callback function for allocating memory blocks when the user-provided allocator is used.

aclError aclrtAllocatorSetAllocFuncToDesc(aclrtAllocatorDesc allocatorDesc, aclrtAllocatorAllocFunc func)

Parameter	Input/Output	Description
allocatorDesc	Input	Pointer to the allocator descriptor. Call aclrtAllocatorCreateDesc to set the allocator description in advance.
func	Input	Callback function for allocating memory blocks. Callback function definition: typedef void (aclrtAllocatorAllocFunc)(aclrtAllocator allocator, size_t size);

Parameter

Input/Output

Description

allocatorDesc

Input

Pointer to the allocator descriptor.

Call aclrtAllocatorCreateDesc to set the allocator description in advance.

func

Input

Callback function for allocating memory blocks.

Callback function definition:

typedef void *(*aclrtAllocatorAllocFunc)(aclrtAllocator allocator, size_t size);

0 on success; otherwise, failure. For details, see aclError.

Parent topic: aclrtAllocatorDesc