Function: allocator_set_free_func_to_desc

Product	Supported (√/x)
Ascend 950PR/Ascend 950DT	√
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas training products	√
Atlas inference products	√
Atlas 200I/500 A2 inference products	√

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

C Prototype

aclError aclrtAllocatorSetFreeFuncToDesc(aclrtAllocatorDesc allocatorDesc, aclrtAllocatorFreeFunc func)

Python Function

ret = acl.rt.allocator_set_free_func_to_desc(allocatorDesc, func)

Parameter	Description
allocatorDesc	Int, pointer address of the allocator descriptor. Call acl.rt.allocator_create_desc to set the allocator description in advance.
func	Callback function for freeing memory blocks. Callback function definition: def allocator_free_func(allocator, block): pass

Return Value	Description
ret	Int, 0 on success; else, failure.

Parent topic: aclrtAllocatorDesc