Function: allocator_set_alloc_func_to_desc

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

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

C Prototype

aclError aclrtAllocatorSetAllocFuncToDesc(aclrtAllocatorDesc allocatorDesc, aclrtAllocatorAllocFunc func)

Python Function

ret = acl.rt.allocator_set_alloc_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 allocating memory blocks. Callback function definition: def allocator_alloc_func(allocator, size): pass

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 definition:

def allocator_alloc_func(allocator, size):
    pass

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

Parent topic: aclrtAllocatorDesc