Mojo Academy · Урок

Переписывание горячего пути на Mojo

Перенесите узкое место с помощью fn и SIMD.

«Переписывание горячего пути на Mojo» — бесплатный урок Mojo Academy на CoddyKit. Это урок 2 из 4. Ты можешь прочитать весь урок бесплатно ниже — а потом практиковать его прямо в браузере с встроенным редактором кода и ИИ-репетитором 24/7. Это часть пути обучения Mojo Academy, и твой прогресс синхронизируется между веб-версией и приложением CoddyKit. Курс Mojo Academy содержит 4 уроков всего.

Части этого урока еще не переведены и отображаются на английском.

Translate the Hot Path

Now take that slow Python loop and rebuild it in Mojo. You keep the same logic but gain real performance from typed, compiled code. 🔧

Reach for fn

Write the kernel as an fn function, not def. Required types let the compiler generate tight machine code for your hot loop.

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

Annotate Every Type

Give each value a concrete type. With no dynamic guessing left, Mojo can optimize the arithmetic the way C would.

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

Port the Loop Body

Copy the inner math one step at a time. Keep the same operations so results match Python exactly before you optimize anything.

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

Verify Correctness First

Run the Mojo version on the same input and compare. Correctness comes before speed, since a fast wrong answer helps no one.

Bring In SIMD

Once it is correct, vectorize. A SIMD value packs several numbers so one instruction processes many elements at once.

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

Process in Vector Chunks

Step through the data in groups equal to the SIMD width. Each pass handles a whole chunk instead of one lonely element.

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

Reduce the Vector

After the loop, collapse the SIMD accumulator into one number with a reduce. That folds every lane down to a single total.

var total = acc.reduce_add()

Handle the Tail

If the length is not a clean multiple of the width, a few elements remain. A small tail loop adds them so nothing is dropped.

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

Re-check the Result

Compare the vectorized output to the Python baseline again. Matching results confirm your SIMD rewrite kept the math honest.

Time the New Kernel

Measure the Mojo kernel on the same data. Even single-threaded, a typed and vectorized kernel usually beats the Python loop by a lot.

Quick Check

Pick the right order for rewriting a hot path in Mojo.

Recap

You ported the loop to a typed fn, verified correctness, then vectorized with SIMD and handled the tail. The hot path is now fast Mojo. 🎯

Часто задаваемые вопросы

Урок «Переписывание горячего пути на Mojo» бесплатный?

Да — полный текст урока «Переписывание горячего пути на Mojo» бесплатно доступен здесь в веб-версии. Чтобы практиковать его интерактивно (встроенный редактор кода и ИИ-репетитор 24/7) и разблокировать остальной курс Mojo Academy, подпишись на CoddyKit PRO. Курс Mojo Academy содержит 4 уроков всего.

Чему я научусь в уроке «Переписывание горячего пути на Mojo»?

Перенесите узкое место с помощью fn и SIMD. Ты практикуешь Mojo Academy с помощью реального кода, который запускаешь прямо в браузере, и ИИ-репетитор 24/7 отвечает на твои вопросы во время урока.

Нужен ли мне опыт, чтобы начать Mojo Academy?

Предыдущий опыт не требуется. Mojo Academy на CoddyKit структурирован для всех уровней — от новичков до продвинутых, поэтому ты можешь начать отсюда или с самого начала и учиться в своем темпе. Это урок 2 из 4.

Сколько времени занимает урок «Переписывание горячего пути на Mojo»?

Большинство уроков CoddyKit занимают около 5–10 минут. Каждый из них компактный и интерактивный, поэтому ты постоянно делаешь прогресс и продолжаешь с того же места в веб-версии и приложении.

Можно ли писать и запускать код в этом уроке Mojo Academy?

Да. Каждый урок Mojo Academy включает встроенный редактор кода, поэтому ты пишешь и запускаешь реальный код прямо в браузере и получаешь моментальную обратную связь от AI — локальная установка не требуется.

Все уроки этого курса

← Назад к Mojo Academy