有人说,我做一款AI芯片,只要原生支持PyTorch就好了呀,用户直接用PyTorch跑模型,根本接触不到CUDA呀。
没错,一般的用户只要在PyTorch层面做应用,但是总是有新的模型架构出来,这些架构都需要做特定的性能优化才能在一个芯片上得到较高的性能,这时候就涉及到算子开发了。
比如说一开始LLM在GPU上的性能不好,后来社区针对Nvidia GPU做了flash attention等的优化才把LLM的性能提升到了比较可观的程度。
CUDA已经被各种开源AI框…。
有人说,我做一款AI芯片,只要原生支持PyTorch就好了呀,用户直接用PyTorch跑模型,根本接触不到CUDA呀。
没错,一般的用户只要在PyTorch层面做应用,但是总是有新的模型架构出来,这些架构都需要做特定的性能优化才能在一个芯片上得到较高的性能,这时候就涉及到算子开发了。
比如说一开始LLM在GPU上的性能不好,后来社区针对Nvidia GPU做了flash attention等的优化才把LLM的性能提升到了比较可观的程度。
CUDA已经被各种开源AI框…。
给小孩Youtube Kids,陪他玩Minecraft,他...
2025-06-20阅读全文 >>买啊,结婚前我说我要买相机,老婆不答应,我就自己买了一个二手...
2025-06-20阅读全文 >>FrontPage是被微软自己淘汰的,不是被市场淘汰的。 ...
2025-06-20阅读全文 >>干得太漂亮了,特拉维夫的制空权都被伊朗夺取了,三分之二以上的...
2025-06-20阅读全文 >>公共DNS算电信业务,自建且不备案就是违规经营电信业务电信业...
2025-06-20阅读全文 >>