Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
mlaprolog v2在tokenx传入null时会导致coredump
Ascend910b
mlaprolog v2在tokenx传入null时正常报错，而不是coredump
In file included from ../attention/mla_prolog_v2/examples/test_aclnn_mla_prolog_v2_nq_bsh.cpp:25: /cann/cann_0804/cann-9.1.0/include/aclnnop/aclnn_mla_prolog_v2_weight_nz.h:56:1: note: declared here 56 | aclnnMlaPrologV2WeightNz(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, const aclrtStream stream); | ^~~~~~~~~~~~~~~~~~~~~~~~ build.sh: line 598: 48423 Segmentation fault ./test_aclnn_${EXAMPLE_NAME}
/assign @wangss21
Thanks for sending an issue! Please fill in the following template to help quickly solve your problem.
Describe the current behavior / 问题描述 (Mandatory / 必填)
mlaprolog v2在tokenx传入null时会导致coredump
Environment / 环境信息 (Mandatory / 必填)
Ascend910b
Steps to reproduce the issue / 重现步骤 (Mandatory / 必填)
mlaprolog v2在tokenx传入null时会导致coredump
Describe the expected behavior / 预期结果 (Mandatory / 必填)
mlaprolog v2在tokenx传入null时正常报错，而不是coredump
Related log / screenshot / 日志 / 截图 (Mandatory / 必填)
In file included from ../attention/mla_prolog_v2/examples/test_aclnn_mla_prolog_v2_nq_bsh.cpp:25:
/cann/cann_0804/cann-9.1.0/include/aclnnop/aclnn_mla_prolog_v2_weight_nz.h:56:1: note: declared here
56 | aclnnMlaPrologV2WeightNz(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, const aclrtStream stream);
| ^~~~~~~~~~~~~~~~~~~~~~~~
build.sh: line 598: 48423 Segmentation fault ./test_aclnn_${EXAMPLE_NAME}
Special notes for this issue/备注 (Optional / 选填)