深度解读昇腾CANN小shape算子计算优化技术,进一步减少调度开销

💡 原文中文,约1800字,阅读约需5分钟。
📝

内容提要

本文介绍了昇腾CANN小shape算子计算优化技术,通过将小shape算子保留在Host侧执行,减少调度开销带来的性能影响。优化后,模型执行性能显著提高。

Q&A

昇腾CANN小shape算子计算优化技术的主要目标是什么?

主要目标是减少调度开销带来的性能影响。

在Host调度模式下,算子的执行单元是如何划分的?

算子的执行单元分为Host CPU执行和Device执行两类。

为什么小shape算子的调度开销需要优化?

因为小shape算子的调度开销往往大于实际计算开销。

优化后,模型的E2E执行时间有何变化?

E2E执行时间从3毫秒减少到约10微秒。

以LLaMA2模型为例,优化后的吞吐量提升了多少?

吞吐提升了5%。

GE是如何识别并优化小shape算子的?

GE识别小shape算子并将其保留在Host侧执行,以减少调度开销。

🏷️

标签

➡️

继续阅读