专栏底层软件编译hbm的时候batch_size=1 可以编译成功, batch_size=2的时候编译失败。

编译hbm的时候batch_size=1 可以编译成功, batch_size=2的时候编译失败。

已解决
zhaokaopeng2023-06-08
91
4

编译hbm的时候batch_size=1 可以编译成功, batch_size=2的时候编译失败。

RuntimeError: HBDK model compilation FAIL, please check with HBDK Group

底层软件
征程5
评论1
0/1000
  • 颜值即正义
    Lv.2

    请问您这边是如何配置的编译参数呢?以及编译命令是?

    2023-06-08
    0
    3
    • zhaokaopeng回复颜值即正义:

      python3 /horizon_bev/tools/compile_perf.py --config configs/segmentation/config.py --out-dir ./hbm_batch2 --opt 3

      deploy_inputs = dict(img=torch.randn((2, 3, 640, 1280)))
      compile_dir = os.path.join(ckpt_dir, "compile")
      compile_cfg = dict(
      march=march,
      name=task_name + "_model",
      out_dir=compile_dir,
      hbm=os.path.join(compile_dir, "model.hbm"),
      layer_details=True,
      input_source=["pyramid"],
      extra_args=["--skip-check"],
      opt="03",
      )
      2023-06-09
      0
    • 颜值即正义回复zhaokaopeng:

      麻烦您提供一下完整的报错截图

      2023-06-12
      0
    • 颜值即正义回复颜值即正义:
      如果在hbm_batch2文件夹下生成了.hbir文件,麻烦提供给我们
      2023-06-12
      0