---
title: 函数：malloc_for_task_scheduler
description: "| 产品 | 是否支持 |"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/API/runtimeapi/aclpythondevg_01_1060.html
sourcePath: /source/zh/canncommercial/900/API/runtimeapi/aclpythondevg_01_1060.html
indexId: 66bf24f99ad5f7cf7b307b55db12c2de63ec1fc00362035d94179cc76cfe035571
---
# 函数：malloc_for_task_scheduler

#### 产品支持情况

| 产品 | 是否支持 |
| --- | --- |
| Atlas 350 加速卡 | √ |
| Atlas A3 训练系列产品 / Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品 / Atlas A2 推理系列产品 | √ |
| Atlas 训练系列产品 | √ |
| Atlas 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | √ |


#### 功能说明

申请AI处理器上Task调度器可使用的内存。

图模式下有部分算子需要使用该类型的内存。


#### 函数原型

- C函数原型
  1 aclError aclrtMallocForTaskScheduler(void **devPtr, size_t size, aclrtMemMallocPolicy policy, aclrtMallocConfig *cfg)

- python函数
  1 dev_ptr, ret = acl.rt.malloc_for_task_scheduler(size, policy, cfg)


#### 参数说明

| 参数名 | 说明 |
| --- | --- |
| size | int，申请内存的大小，单位Byte。size不能为0。 |
| policy | int，内存分配规则。 若配置的内存分配规则超出aclrtMemMallocPolicy取值范围，size≥2M时，按大页申请内存，否则按普通页申请内存。 |
| cfg | dict，内存配置信息。不指定配置时，此处可传空字典，具体请参见aclrtMallocConfig。 |


#### 返回值说明

| 返回值 | 说明 |
| --- | --- |
| devPtr | int，指向Device上已分配内存的指针地址。 |
| ret | int，错误码，返回0表示成功，返回其它值表示失败。 |