aclrtAllocatorSetAllocAdviseFuncToDesc

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 allocating memory blocks based on recommended addresses when the user-provided allocator is used. This function is generally used in memory overcommitment scenario.

Prototype

aclError aclrtAllocatorSetAllocAdviseFuncToDesc(aclrtAllocatorDesc allocatorDesc, aclrtAllocatorAllocAdviseFunc 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 allocating memory blocks based on the recommended addresses. Callback function definition: typedef void (aclrtAllocatorAllocAdviseFunc)(aclrtAllocator allocator, size_t size, aclrtAllocatorAddr addr);

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 based on the recommended addresses.

Callback function definition:

typedef void *(*aclrtAllocatorAllocAdviseFunc)(aclrtAllocator allocator, size_t size, aclrtAllocatorAddr addr);

Returns

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

Parent topic: aclrtAllocatorDesc