专栏算法工具链我在J6M上进行模型验证,模型编译出现了问题,提示会报这个错误,能帮忙解决下吗

我在J6M上进行模型验证,模型编译出现了问题,提示会报这个错误,能帮忙解决下吗

已解决
Yang~2025-04-03
104
4

我在附件中提供了最小模型,模型编译会报如下错误:

2025-04-03 02:37:37,784 INFO Start to compile model. loc(fused<#hbdk.track], originalAlignOutputsShape = [<1, 1000, 8, 4, 4, 32>]>>["GatherElements_218"]): error: 'hbir.gather_elements' op codegen failed (layer[4] in 7 layers) notes: ['see current operation: %185 = "hbir.gather_elements"(%41, %184) <{dim = 1 : i64, extra = #hbdk.extra (d0, d1, d2, d3, d4, d5)], skipCheck = [false], attr = {layer_id = 186 : i64}>}> : (tensor<1x26550x8x4x4x32xsi8, {block = "4W_32C", order = [0, 4, 5, 1, 3, 2]}>, tensor<1x1000x8x4x4x32xsi64, {block = "16C", order = [0, 3, 4, 1, 2, 5]}>) -> tensor<1x1000x8x4x4x32xsi8, {block = "4W_32C", order = [0, 4, 5, 1, 2, 3]}>'] MLIR Python Diagnostic handler raised exception: Exception: error detected At: /usr/local/lib/python3.10/dist-packages/hbdk4/compiler/utils/default.py(18): callback /usr/local/lib/python3.10/dist-packages/hbdk4/compiler/apis.py(237): compile /usr/local/lib/python3.10/dist-packages/hbdk4/compiler/utils/default.py(39): wrapper /usr/local/lib/python3.10/dist-packages/horizon_tc_ui/hbir_handle.py(76): compile_model /usr/local/lib/python3.10/dist-packages/horizon_tc_ui/utils/wrap_utils.py(49): wrapper /usr/local/lib/python3.10/dist-packages/horizon_tc_ui/compile/hbm_builder.py(271): compile_model /usr/local/lib/python3.10/dist-packages/horizon_tc_ui/compile/hbm_builder.py(295): build /usr/local/lib/python3.10/dist-packages/horizon_tc_ui/hb_compile.py(130): hbm_build /usr/local/lib/python3.10/dist-packages/horizon_tc_ui/hb_compile.py(215): fast_perf_mode /usr/local/lib/python3.10/dist-packages/horizon_tc_ui/hb_compile.py(301): main /usr/local/lib/python3.10/dist-packages/horizon_tc_ui/utils/tool_utils.py(135): __decorator /usr/local/lib/python3.10/dist-packages/click/core.py(610): invoke /usr/local/lib/python3.10/dist-packages/click/core.py(1066): invoke /usr/local/lib/python3.10/dist-packages/click/core.py(782): main /usr/local/lib/python3.10/dist-packages/click/core.py(829): __call__ /usr/local/bin/hb_compile(8): loc(fused<#hbdk.track], originalAlignOutputsShape = [<1, 1000, 8, 4, 4, 32>]>>["GatherElements_218"]): error: 'hbir.gather_elements' op codegen failed (layer[4] in 7 layers) terminate called without an active exception PLEASE submit a bug report to https://github.com/llvm/llvm-project/issues/ and include the crash backtrace. Stack dump: 0. Program arguments: Stack dump without symbol names (ensure you have llvm-symbolizer in your PATH or set the environment var `LLVM_SYMBOLIZER_PATH` to point to it): 0 libHBDKPythonCAPI.so 0x00007a70a963eeeb

 

 

附件:
算法工具链
征程6
评论3
0/1000
  • Huanghui
    Lv.5

    你好,这段报错代码可能是 张量形状不匹配、数据排布格式错误,还有数据类型不同 。 输入1 (1x26550x8x4x4x32) 和 输入2 (1x1000x8x4x4x32) 维度不匹配 。 block 格式不一致:输入1是 "4W_32C",输入2是 "16C" 。 索引张量的数据类型 si64 可能与 si8 不兼容

    2025-04-03
    1
    0
  • Jade-self
    Lv.5

    可以找你们的项目经理,请他帮你更新 工具链到OE3.0.31,使用该环境是可以正常编译你附件中提供的test.onnx模型的。

    2025-04-06
    1
    1
    • Yang~回复Jade-self:

      是的,使用这个版本可以正常编译了

      2025-04-07
      0
  • 费小财
    Lv.5

    看上去是输入的维度异常了

    2025-04-07
    0
    0