aclrtAllocatorSetFreeFuncToDesc

Applicability

Product

Supported

Atlas A3 training products/Atlas A3 inference products

√

Atlas A2 training products/Atlas A2 inference products

√

Atlas 200I/500 A2 inference products

√

Atlas inference products

√

Atlas training products

√

Description

Sets the callback function for releasing memory blocks when the user-provided Allocator is used.

Prototype

aclError aclrtAllocatorSetFreeFuncToDesc(aclrtAllocatorDesc allocatorDesc, aclrtAllocatorFreeFunc func)

Parameters

Parameter

Input/Output

Description

allocatorDesc

Input

Pointer to the Allocator descriptor.

Call the aclrtAllocatorCreateDesc API to set the Allocator description in advance.

func

Input

Callback function for freeing memory blocks.

Callback function definition:

typedef void (*aclrtAllocatorFreeFunc)(aclrtAllocator allocator, aclrtAllocatorBlock block);

Returns

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