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

533次阅读
没有评论

多元共进|2023
Google 开发者大会 精彩演讲回顾
将 Python 程序移植到 Mojo,性能提升 250 倍、速度比 C 还快

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

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

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

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

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

详情

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

Read More 

正文完
可以使用微信扫码关注公众号(ID:xzluomor)
post-qrcode
 
评论(没有评论)
Generated by Feedzy