aclrtAllocatorSetFreeFuncToDesc

Applicability

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

Description

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

Prototype

1
aclError aclrtAllocatorSetFreeFuncToDesc(aclrtAllocatorDesc allocatorDesc, aclrtAllocatorFreeFunc func)

Parameters

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 releasing memory blocks.

Callback function definition:

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

Returns

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