有人说,我做一款AI芯片,只要原生支持PyTorch就好了呀,用户直接用PyTorch跑模型,根本接触不到CUDA呀。
没错,一般的用户只要在PyTorch层面做应用,但是总是有新的模型架构出来,这些架构都需要做特定的性能优化才能在一个芯片上得到较高的性能,这时候就涉及到算子开发了。
比如说一开始LLM在GPU上的性能不好,后来社区针对Nvidia GPU做了flash attention等的优化才把LLM的性能提升到了比较可观的程度。
CUDA已经被各种开源AI框…。
为什么程序员独爱用Mac进行编程?…
天涯论坛关闭后,除了知乎,大家都在逛什么?…
如何看待盲人 UP 主蒙眼炒鸡蛋在成都旅游期间,要求各大医院无处方买胰岛素,最后进抢救室?…
如何才能在少掉肌肉的同时尽快减脂?…
备案号:豫-ICP备11334303号-1 网站地图