logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

asnumpy:让你的 NumPy 代码零改动跑在昇腾 NPU 上

有个同事之前跟我说,他写了一套数据预处理的 pipeline,全是 NumPy 写的,后来要迁移到昇腾 NPU 上跑,“感觉天都要塌了”——几千行 NumPy 代码,难道要全部重写?后来他发现了 asnumpy,这个东西基本上解决了这个问题。不需要改你的 NumPy 代码,直接换一下 import,数据搬上 NPU 跑,原来的语法一个不动。

文章图片
#numpy#人工智能
asnumpy:让你的 NumPy 代码零改动跑在昇腾 NPU 上

有个同事之前跟我说,他写了一套数据预处理的 pipeline,全是 NumPy 写的,后来要迁移到昇腾 NPU 上跑,“感觉天都要塌了”——几千行 NumPy 代码,难道要全部重写?后来他发现了 asnumpy,这个东西基本上解决了这个问题。不需要改你的 NumPy 代码,直接换一下 import,数据搬上 NPU 跑,原来的语法一个不动。

文章图片
#numpy#人工智能
FlashAttention 在昇腾 NPU 上是怎么省显存的?我花了两天把它搞明白了

FlashAttention 在昇腾 NPU 上是怎么省显存的?我花了两天把它搞明白了

文章图片
#android#人工智能
FlashAttention 在昇腾 NPU 上是怎么省显存的?我花了两天把它搞明白了

FlashAttention 在昇腾 NPU 上是怎么省显存的?我花了两天把它搞明白了

文章图片
#android#人工智能
到底了