题目内容

在深度学习系统中，算法由若干基本计算单元构成，这些基本单元称为算子。其中负责矢量运算的算子称为矢量算子。NPU 内部包含矩阵计算单元和向量计算单元，二者都可以执行矢量算子，并且能够并行工作、互不阻塞。不过，两类单元的计算效率不同：若某个矢量算子在矩阵计算单元上的运行时间为 $N$，则其在向量计算单元上的运行时间为 $6N$。

现给出一组矢量算子。每个算子既可以部署到矩阵计算单元，也可以部署到向量计算单元。你需要为每个算子指定一个执行单元，使得两个单元上的总耗时同时开始后，完成全部算子的时间尽可能短。由于两个单元并行执行，整体完成时间等于矩阵计算单元总耗时与向量计算单元总耗时的最大值。

为了简化计算，本题约定：

• 单个算子只能部署到矩阵计算单元或向量计算单元中的一个，不能拆分。
• 部署到向量计算单元的算子，在原始输入顺序中必须形成一个连续区间；也就是说，如果选择某些算子放到向量单元，它们在给定序列中必须是紧挨着的一段。

开通会员即可查看完整视频题解： 1.题目讲解 2.思路分析 3.逐行代码手写