有人说,我做一款AI芯片,只要原生支持PyTorch就好了呀,用户直接用PyTorch跑模型,根本接触不到CUDA呀。
没错,一般的用户只要在PyTorch层面做应用,但是总是有新的模型架构出来,这些架构都需要做特定的性能优化才能在一个芯片上得到较高的性能,这时候就涉及到算子开发了。
比如说一开始LLM在GPU上的性能不好,后来社区针对Nvidia GPU做了flash attention等的优化才把LLM的性能提升到了比较可观的程度。
CUDA已经被各种开源AI框…。
6月17日我去北京出差,出租车司机告诉我,北京今年要阅兵。 ...
因为,Windows 的兼容性真的太好了。 举个例子: 十...
目前来看不会比24日的更高,但是也是超保水位的大洪水。 ...
普通用户很难看出来的。 利益相关,我业余工作就是制作散片cp...
我同事跟我吐槽她婆婆: 每天晚上洗完澡,不穿睡衣,用个浴巾裹...
去闲鱼上搜 MacBook Pro M1,都是某互联网大厂满...