如何在gem5中配置矩阵乘法器

Question

我们计划使用gem5模拟Google的Tensor处理单元（TPU）。我们不确定从gem5开始。我们已经按照学习文件夹中的所有教程进行了学习。首先，我们要实现矩阵乘法器。我们该怎么做？

Answer 1

如果我理解正确，您想向gem5添加一个新设备，该设备可以进行矩阵乘法。

我还没有看到关于将设备添加到gem5的任何优秀教程。当我学习它时，我会写一个。

因此，最好的选择是查看现有的gem5源代码，在src/dev下查看，并尝试查找与您的设备最相似的设备。

设备/ gem5界面应该不太难：您基本上只需要了解如何执行以下操作即可：

• 注册
• 中断
• DMA

然后，我建议您使用以下方法进行测试：

• 裸机程序。 Here is a possible setup。这些使您可以跳过启动，这是一个巨大的胜利。
• 一个Linux内核模块，当您感觉快完成时可以实际运行内容

也许您可以从有关QEMU的答案中得到一些启发：How to add a new device in QEMU source code?