---
title: mem_pool_cfg
description: "| 产品 | 是否支持 |"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/API/hixlug/llmprv2_42_0080.html
sourcePath: /source/zh/canncommercial/900/API/hixlug/llmprv2_42_0080.html
indexId: c3314a9a44a4c41fd56e9f92bba3fd9704b41e41dd7894bf3feae4951690a9bf61
---
# mem_pool_cfg

#### 产品支持情况

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


说明：针对 Atlas A2 训练系列产品 / Atlas A2 推理系列产品 ，仅支持Atlas 800I A2 推理服务器、A200I A2 Box 异构组件。


#### 函数功能

配置内存池相关配置项。


#### 函数原型

```
mem_pool_cfg(mem_pool_cfg)
```


#### 参数说明

| 参数名称 | 数据类型 | 取值说明 |
| --- | --- | --- |
| mem\_pool\_cfg | string | json数组格式字符串，包含memory\_size，page\_shift。 |


| 配置项 | 可选/必选 | 描述 |
| --- | --- | --- |
| memory\_size | 必选 | 当前内存池的大小，类型为int，大于0，单位为Byte。 |
| page\_shift | 可选 | page\_size的位移量，用于计算page\_size。分配内存时会对齐到page\_size的倍数，需要根据实际场景设置为合适的大小。 类型为int，取值范围为[10, 31)。 例如：page\_shift = 16时，page\_size为1<<16=65536。 默认值为16。 |


#### 调用示例

```
from llm_datadist import LLMConfig
llm_config = LLMConfig()
llm_config.mem_pool_cfg= "{\"memory_size\": 18737418240, \"page_shift\": 16}"
```


#### 返回值

正常情况下无返回值。

参数错误可能抛出TypeError或ValueError。


#### 约束说明

无