将Python程序移植到Mojo,性能提升250倍、速度比C版本快20%
出品 | OSC开源社区(ID:oschina2013)
Aydyn Tairov 是一名开源作者,也是 Meta 前工程师,他此前将 GitHub 上火热的纯 C 语言实现的 llama2.c
项目移植到了 Python ——llama2.py
。
上周 Mojo 编程语言正式开放下载,并且声称比 Python 快 68000 倍。
Mojo 是 Modular AI 公司开发的新编程语言,它结合了 Python 的易用性以及 C 语言的可移植性和性能,目标是使其成为 AI 研究和生产的理想选择。
于是 Aydyn Tairov 马不停蹄地就开始将 llama2.py
移植到 Mojo——llama2.mojo
,结果非常出乎意料。
他表示,Mojo SIMD 原语帮助将 Python 的糟糕性能提升了近 250 倍。此外,由于他使用了 matmul 的矢量化助手,经测试 llama2.py
速度比 C 版本快 20%。
https://twitter.com/tairov/status/1701194900228764023
往期推荐
点这里 ↓↓↓ 记得 关注✔ 标星⭐ 哦
微信扫码关注该文公众号作者
戳这里提交新闻线索和高质量文章给我们。
来源: qq
点击查看作者最近其他文章