Describe the current behavior / 问题描述 (Mandatory / 必填)

generate多卡推理和pipeline多卡推理失败，报如下错误：
TypeError: For Operator[ReshapeAndCache], slot_mapping's type 'None' does not match expected type 'Tensor'.

Environment / 环境信息 (Mandatory / 必填)

• Hardware Environment(Ascend/GPU/CPU) / 硬件环境:

Please delete the backend not involved / 请删除不涉及的后端:
/device ascend

• Software Environment / 软件环境 (Mandatory / 必填):
  -- MindSpore version (e.g., 1.7.0.Bxxx) :
  -- Python version (e.g., Python 3.7.5) :
  -- OS platform and distribution (e.g., Linux Ubuntu 16.04):
  -- GCC/Compiler version (if compiled from source):
  ms版本：commit_id = '[sha1]:84aa3010,[branch]:(HEAD,origin/master,origin/HEAD,master)'
  run包：runpkg_version:Milan_C17/20240414
  mf版本： commit de0dda0752cbee19900e1a9116dcb4924bb1b41e (HEAD -> dev, origin/dev, origin/HEAD)

• Excute Mode / 执行模式 (Mandatory / 必填)(PyNative/Graph):

Please delete the mode not involved / 请删除不涉及的模式:
/mode pynative
/mode graph

Related testcase / 关联用例 (Mandatory / 必填)

不涉及

Steps to reproduce the issue / 重现步骤 (Mandatory / 必填)

不涉及

执行命令：
source /usr/local/Ascend/latest/bin/setenv.bash
source /home/miniconda3/bin/activate ci
export ASCEND_HOME_PATH=/usr/local/Ascend/latest
export PYTHONPATH=/home/jenkins0/mindformers/:$PYTHONPATH

bash scripts/msrun_launcher.sh "predict_custom.py
--yaml_file path/to/config_yaml
--checkpoint_path path/to/shard_checkpoint_dir
--model_type llama2_70b" 4

Describe the expected behavior / 预期结果 (Mandatory / 必填)

正常推理

Related log / screenshot / 日志 / 截图 (Mandatory / 必填)

报错日志：
Traceback (most recent call last):
File "predict_custom_generate.py", line 83, in
main(args)
File "predict_custom_generate.py", line 53, in main
warm_up_model.infer_predict_layout(ms.Tensor(np.ones(shape=(model_config.batch_size, model_config.seq_length)), ms.int32))
File "/home/miniconda3/envs/ci/lib/python3.7/site-packages/mindspore/train/model.py", line 1896, in infer_predict_layout
predict_net.compile(*predict_data)
File "/home/miniconda3/envs/ci/lib/python3.7/site-packages/mindspore/nn/cell.py", line 997, in compile
jit_config_dict=self._jit_config_dict, **kwargs)
File "/home/miniconda3/envs/ci/lib/python3.7/site-packages/mindspore/common/api.py", line 1642, in compile
result = self._graph_executor.compile(obj, args, kwargs, phase, self._use_vm_mode())
TypeError: For Operator[ReshapeAndCache], slot_mapping's type 'None' does not match expected type 'Tensor'.
The reason may be: lack of definition of type cast, or incorrect type when creating the node.

• Framework Unexpected Exception Raised:

This exception is caused by framework's unexpected error. Please create an issue at https://gitee.com/mindspore/mindspore/issues to get help.

• C++ Call Stack: (For framework developers)

mindspore/ccsrc/pipeline/jit/ps/static_analysis/prim.cc:1263 CheckArgsSizeAndType

Special notes for this issue/备注 (Optional / 选填)

走给 谭伟城