将 Python 程序移植到 Mojo,性能提升 250 倍、速度比 C 还快

Aydyn Tairov 是一名开源作者,也是 Meta 前工程师,他此前将 GitHub 上火热的纯 C 语言实现的llama2.c项目移植到了 Python ——llama2.py

上周 Mojo 编程语言正式开放下载,并且声称比 Python 快 68000 倍。

于是 Aydyn Tairov 马不停蹄地就开始将llama2.py移植到 Mojo——llama2.mojo,结果非常出乎意料。

他表示,Mojo SIMD 原语帮助将 Python 的糟糕性能提升了近 250 倍。此外,由于他使用了 matmul 的矢量化助手,经测试llama2.py速度比 C 版本快 20%。

详情

https://github.com/tairov/llama2.mojo
https://twitter.com/tairov/status/1701194900228764023

猜你喜欢

转载自www.oschina.net/news/257797/llama2-mojo-faster-than-c