aclrtMallocForTaskScheduler

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	√

Allocates the memory that can be used by the task scheduler on the Ascend AI Processor.

Some operators in graph mode need to use the memory of this type.

aclError aclrtMallocForTaskScheduler(void **devPtr, size_t size, aclrtMemMallocPolicy policy, aclrtMallocConfig *cfg)

Parameter	Input/Output	Description
devPtr	Output	Pointer to the pointer to the allocated device memory.
size	Input	Requested allocation size in bytes. The size cannot be 0.
policy	Input	Memory allocation policy. If the configured memory allocation policy is not within the value range of aclrtMemMallocPolicy, and the size is greater than or equal to 2 MB, the huge page memory is allocated; otherwise, the common page memory is allocated.
cfg	Input	Memory configuration. If no configuration is specified, NULL can be passed.

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

Parent topic: Memory Management