將 Python 程序移植到 Mojo,性能提升 250 倍、速度比 C 還快
Aydyn Tairov 是一名開源作者,也是 Meta 前工程師,他此前將 GitHub 上火熱的純 C 語言實(shí)現(xiàn)的 llama2.c項(xiàng)目移植到了 Python ——llama2.py。
上周 Mojo 編程語言正式開放下載,并且聲稱比 Python 快 68000 倍。
于是 Aydyn Tairov 馬不停蹄地就開始將 llama2.py 移植到 Mojo——llama2.mojo,結(jié)果非常出乎意料。
他表示,Mojo SIMD 原語幫助將 Python 的糟糕性能提升了近 250 倍。此外,由于他使用了 matmul 的矢量化助手,經(jīng)測試 llama2.py 速度比 C 版本快 20%。
詳情
https://github.com/tairov/llama2.mojohttps://twitter.com/tairov/status/1701194900228764023