This thread has been locked.

If you have a related question, please click the "Ask a related question" button in the top right corner. The newly created question will be automatically linked to this question.

[参考译文] AM69A:DSP 自定义操作:输出 Tensor 具有 4D 形状时输出值不正确

Guru**** 2439670 points


请注意,本文内容源自机器翻译,可能存在语法或其它翻译错误,仅供参考。如需获取准确内容,请参阅链接中的英语原文或自行翻译。

https://e2e.ti.com/support/processors-group/processors/f/processors-forum/1543417/am69a-dsp-custom-op-incorrect-output-values-when-output-tensor-has-4d-shape

器件型号:AM69A


工具/软件:

TI 团队大家好、

我正在实施自定义运算符、并且在上运行推理时遇到可重现的问题 来实现

问题总结:

  • 自定义 op 输出张量 具有 4D 形状(例如)[1, 4, 3, 4]
  • 当我为invoke函数内部的输出(在 DSP 上)分配值时、i 在分配后立即打印值 它们出现了 100%正确
  • 然而,运算符运行后,模型的最终输出(例如,我从推理结果获得的内容)包含 值不正确或损坏 —它们与我在分配过程中打印的内容不匹配。
  • 相同的定制操作码 、形状相同、运行 完美地放置在 PC 上 如何使用 TIDL 推理。

其他意见:

  • 为了隔离问题,我从自定义操作中删除所有逻辑,然后用常量值填充输出(例如,)output[i] = 10。 即使在这种情况下、DSP 输出仍然不正确。
  • 出现问题 仅当输出张量有 4 个维度时 和时 通道尺寸(第二或第三索引) 大于 2 时才有意义。
  • 这表明问题是这样的 但它不在计算中 、但可能是 DSP 运行时处理多维输出缓冲区的方式。

问题:

  • 在自定义 DSP 操作中写入 4D 输出张量是否有任何已知问题或限制?

非常感谢您对调试该问题的支持。

此致、

  • 请注意,本文内容源自机器翻译,可能存在语法或其它翻译错误,仅供参考。如需获取准确内容,请参阅链接中的英语原文或自行翻译。

    您好;

    感谢您问这个问题。  

    您能否共享程序文件(没有内部逻辑是可以的,只是有常量值的输出)? 以及编译和运行的步骤/命令。 我们将尝试在我们这边复制您的操作。

    此致

    Wen Li