#### 1.问题描述

> Llama2网络，pretrain_llama2_7b.yaml预训练脚本，num_layers:16，8卡。FlopsUtilizationCollector CallBack和sink_size: 1 同时配置时报错，训练报 RuntimeError: Exception thrown from dataset pipeline. Refer to 'Dataset Pipeline Error Message'.

#### 2.环境信息

- **硬件环境**:

| 后端类型 | 硬件具体类别 |

| -------- | ------------ |

| Ascend | 910B4 |

| CPU | Kunpeng-920 |

- **软件环境**:

| Software | Version |

| -------------------- | ------------------------------------------------- |

| MindSpore | 2.4.1 |

| MindFormers | 1.3.0 |

| CANN | 8.0.RC3.alpha002 |

| Python | 3.9.5 |

| OS platform | Linux 4.19.90-2107.6.0.0192.8.oe1.bclinux.aarch64 |

| GCC/Compiler version | 7.3.0 |

#### 3.关联用例

> - **用例1**: sink_size: 2，无FlopsUtilizationCollector CallBack

> 目的：验证默认配置是否可以正常拉起Llama2训练。

>

> - **用例2**: sink_size: 1，无FlopsUtilizationCollector CallBack

>

> 目的：验证ink_size: 1，无FlopsUtilizationCollector CallBack是否可以正常拉起Llama2训练。

>

> - **用例3**: sink_size: 2，无FlopsUtilizationCollector CallBack

>

> 目的：验证ink_size: 2，有FlopsUtilizationCollector CallBack是否可以正常拉起Llama2训练。

> - **用例4**: sink_size: 1，配置FlopsUtilizationCollector CallBack

>

> 目的：验证ink_size: 1，配置FlopsUtilizationCollector CallBack是否可以正常拉起Llama2训练。

>

> ```yaml

> sink_size: 1

>

> - type: FlopsUtilizationCollector

> data_size: 1

> computility: XXXXX

> full_flops: True

> ```

#### 4.Steps to reproduce the issue / 重现步骤

> （1）cd mindformers

> （2）bash scripts/msrun_launcher.sh "run_mindformer.py \

> --config configs/llama2/pretrain_llama2_7b.yaml \

> --pretrain_dataset/openwebtext/openwebtext4096.mindrecord \

> --use_parallel True \

> --run_mode 'train'" \

> 8

> （3）验证是否能够正常拉起训练。

#### 5.Describe the expected behavior / 预期结果

> **【预期结果】**：用例1、2、3能够正常拉起训练，用例4无法正常持续训练，报 RuntimeError: Exception thrown from dataset pipeline. Refer to 'Dataset Pipeline Error Message'.

#### 6.Related log / screenshot / 日志 / 截图

用例4报错关键日志截图：