Mojo Academy · 课时

用 Mojo 重写热点路径

Q: 「用 Mojo 重写热点路径」这节课中我会学到什么？

使用 fn 和 SIMD 转换瓶颈部分 你通过在浏览器中直接运行的动手代码来练习 Mojo Academy，全天候 AI 导师会在你学习这节课的过程中回答你的问题。

使用 fn 和 SIMD 转换瓶颈部分

第 2 / 4 课13 个步骤

用 Mojo 重写热点路径 是 CoddyKit 上的免费 Mojo Academy 课时。这是第 2 节课，共 4 节。你可以在下方免费阅读本课时的完整内容 — 然后在浏览器中使用内置代码编辑器和全天候 AI 导师进行实践。这是 Mojo Academy 学习路径的一部分，你的进度在网页和 CoddyKit 应用中同步。 Mojo Academy 课程共包含 4 节课。

移植热点路径

现在将那段缓慢的 Python 循环用 Mojo 重写。您可以保留相同的逻辑，同时从带类型的编译代码中获得真正的性能。🔧

使用 fn

将核心代码写成 fn 函数，而不是 def。必需的类型可以让编译器为热点循环生成紧凑的机器代码。

fn sum_squares(data: List[Float64]) -> Float64:
    var total: Float64 = 0.0
    return total

为每种类型添加注解

为每个值指定具体的类型。不再需要动态猜测后，Mojo 就能像 C 一样优化算术运算。

var total: Float64 = 0.0
var n: Int = len(data)

移植循环体

一次一步地复制内层数学运算。在优化之前保持相同的操作，确保结果与 Python 完全一致。

for i in range(len(data)):
    total += data[i] * data[i]

先验证正确性

使用相同的输入运行 Mojo 版本并进行比较。正确性优先于速度，因为错误但快速的答案对任何人都没有帮助。

引入 SIMD

确认正确后，再进行向量化。一个SIMD值可以打包多个数字，让一条指令同时处理多个元素。

alias width = 4
var acc = SIMD[DType.float64, width](0)

按向量块处理

按照等于 SIMD宽度的组遍历数据。每次处理一个完整的数据块，而不是孤零零的单个元素。

for i in range(0, n, width):
    var chunk = data.load[width](i)
    acc += chunk * chunk

归约向量

循环结束后，使用归约将 SIMD 累加器折叠为一个数字。这样可以将每个通道合并为一个总数。

var total = acc.reduce_add()

处理尾部元素

如果长度不是宽度的整数倍，就会剩下几个元素。一个小型的尾部循环会将它们加进去，确保没有数据被丢弃。

for i in range(n - (n % width), n):
    total += data[i] * data[i]

重新检查结果

再次将向量化输出与 Python 基线比较。匹配的结果可以确认您的 SIMD 重写保持了计算的正确性。

测量新的核心代码

使用相同的数据测量 Mojo 核心代码。即使只使用单线程，带类型并经过向量化的核心代码通常也会比 Python 循环快很多。

快速检查

选择在 Mojo 中重写热点路径的正确顺序。

在交互式课程中回答

回顾

您将循环移植到了带类型的fn中，验证了正确性，然后使用 SIMD 进行向量化并处理了尾部元素。热点路径现在已经成为快速的 Mojo 代码。🎯

免费开始

用 AI 导师学习 Mojo — 免费

在浏览器中编写并运行真实代码，获得全天候 AI 导师的即时帮助，并在网页或应用中继续学习。

课程: 30
课程: 120

免费开始学习查看完整课程

常见问题解答

「用 Mojo 重写热点路径」课时是免费的吗？

是的 — 「用 Mojo 重写热点路径」的完整文本可在网页上免费阅读。要进行交互式练习（内置代码编辑器和全天候 AI 导师）并解锁 Mojo Academy 课程的其余内容，请升级到 CoddyKit PRO。 Mojo Academy 课程共包含 4 节课。

「用 Mojo 重写热点路径」这节课中我会学到什么？

使用 fn 和 SIMD 转换瓶颈部分你通过在浏览器中直接运行的动手代码来练习 Mojo Academy，全天候 AI 导师会在你学习这节课的过程中回答你的问题。

学习 Mojo Academy 需要有经验吗？

无需任何先前经验。CoddyKit 上的 Mojo Academy 课程适合初学者到高级学习者，你可以从这里开始或从头开始，按照自己的节奏学习。这是第 2 节课，共 4 节。

「用 Mojo 重写热点路径」课时需要多长时间？

大多数 CoddyKit 课程大约需要 5–10 分钟。每节课都很精短且互动，所以你能稳步进步，并在网页和应用中从离开的地方继续。

我能在这节 Mojo Academy 课中编写并运行代码吗？

能。每节 Mojo Academy 课都包含内置代码编辑器，你可以在浏览器中直接编写并运行真实代码，并获得即时 AI 反馈 — 无需本地设置。

此课程中的所有课时

← 返回 Mojo Academy