deepseek在mindspore上的适配问题,aclnnCumsumGetWorkspaceSize call failed
收藏回复举报
deepseek在mindspore上的适配问题,aclnnCumsumGetWorkspaceSize call failed
t('forum.solved') 已解决
新人帖
发表于2025-07-18 10:07:13
0 查看

在根据课程 https://www.hiascend.com/developer/courses/detail/1925362775376744449 昇思+昇腾开发板:软硬结合玩转DeepSeek开发实战中,在模型适配章节,运行

pytest -v -s /home/HwHiAiUser/.local/lib/python3.9/site-packages/tests/ut/transformers/models/qwen2/test_modeling_qwen2.py::Qwen2ModelTest::test_new_cache_format_0

命令时,出现报错  

self = Prim[CumsumExt], input = Tensor(shape=[7], dtype=Int32, value= [1, 1, 1, 1, 1, 1, 1]), dim = 0, dtype = None 
    def __call__(self, input, dim, dtype=None): 
>         return _convert_stub(pyboost_cumsum_ext(self, [input, dim, dtype if dtype is None else dtype_to_type_id('CumsumExt', 'dtype', dtype)])) 
E         RuntimeError: aclnnCumsumGetWorkspaceSize call failed, please check! 
E          
E         ----------------------------------------------------
E         - Ascend Error Message: 
E         ---------------------------------------------------- 
E         EZ1001: [PID: 4173] 2025-07-18-09:14:56.976.301 support for Ascend310B is not implemented[THREAD:4173] 
E          
E         (Please search "CANN Common Error Analysis" at https://www.mindspore.cn for error code description) 
E          
E         ---------------------------------------------------- 
E         - C++ Call Stack: (For framework developers) 
E         ---------------------------------------------------- 
E         mindspore/ops/kernel/ascend/pyboost/customize/cumsum_ext.cc:47 operator() 
 
.local/lib/python3.9/site-packages/mindspore/ops/auto_generate/gen_ops_prim.py:4971: RuntimeError

我是应该替换pyboost_cumsum_ext吗,但是我好像没有找到pyboost_cumsum_ext的定义

我要发帖子