E-mail : xitang@126.com郑州市紫荆山路72号裕鸿花园D座5层
当前位置: 首页 >

CUDA真是NVIDIA绝对牢不可破的生态吗?

有人说,我做一款AI芯片,只要原生支持PyTorch就好了呀,用户直接用PyTorch跑模型,根本接触不到CUDA呀。

没错,一般的用户只要在PyTorch层面做应用,但是总是有新的模型架构出来,这些架构都需要做特定的性能优化才能在一个芯片上得到较高的性能,这时候就涉及到算子开发了。

比如说一开始LLM在GPU上的性能不好,后来社区针对Nvidia GPU做了flash attention等的优化才把LLM的性能提升到了比较可观的程度。

CUDA已经被各种开源AI框…。

CUDA真是NVIDIA绝对牢不可破的生态吗?
分享到:

喜堂郑州店  TEL:0371-53737377 / 53737317  E-MAIL:xitang@126.com
                ADD:郑州市紫荆山路72号裕鸿花园D座5层
版权@|备案:皖-ICP备83350062号-1|网站地图 织梦模板