aclrtAllocatorSetGetAddrFromBlockFuncToDesc

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	√

Sets the callback function for obtaining the device memory address based on the allocated block when the user-provided Allocator is used.

aclError aclrtAllocatorSetGetAddrFromBlockFuncToDesc(aclrtAllocatorDesc allocatorDesc, aclrtAllocatorGetAddrFromBlockFunc func)

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 obtaining the device memory address based on the allocated block. Callback function definition: typedef void (aclrtAllocatorGetAddrFromBlockFunc)(aclrtAllocatorBlock block);

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 obtaining the device memory address based on the allocated block.

Callback function definition:

typedef void *(*aclrtAllocatorGetAddrFromBlockFunc)(aclrtAllocatorBlock block);

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

Parent topic: aclrtAllocatorDesc