第二百八十七章 视界技能,初试身手 (第2/3页)
,眼看就要崩溃。
这时,韩路一的手指停了下来。
这就是最佳大小了。
在意念世界中,韩路一手掌一挥,这个最佳参数也被写进电脑上的代码里。
易如反掌。
虚拟投影的世界里,韩路一漂浮在半空中,俯视着经过调整、正高速有序运转的机械工厂,心里涌出一种快感。
这种快感,就像把俄罗斯方块堆得高高的,只留边上一条空列,然后来了一个长条一落到底,一下消除四行。
就是那样一种快感。
一种专属于工程师的快感。
韩路一再次打开视界,检查这个算子适配的实现情况。
【当前算子:FusedAttentionBackward】
【优化策略:块内重计算+数据块(Tile)大小校准】
【单步耗时:46.8ms→ 4.6ms】
【显存读写量:理论最优值的1.09倍】
【利用率:31%→ 91%】
【状态:已适配✓】
韩路一看了一眼这组数字,脸上露出一点笑容。
单步耗时缩短到原来的十分之一,利用率从三成拉到九成以上,显存读写也逼近理论极限。
他轻轻吐出一口气。
这才是L100应该有的样子。
做完这个最难算子的适配,韩路一对新研发的视界技能【算子适配】有了大概了解。
网上所有能找到的硬件、编译、调优知识,视界全都吸收了;最终做出来的这个视界技能,更是把工程验证过程直接变成了意念里的沙盒游戏。
至于效率嘛……大概是一个大型工程师团队的几千倍以上。
竟然如此强大。
视哥,不差。
韩路一正准备一鼓作气接着做下一个算子,只需退出现在加载的【注意力融合反向算子】模块,再载入——
突然,他眼前充满科幻色彩的半透明界面和虚拟硬件模型闪烁两下,消失了。
韩路一错愕地眨了眨眼,意识到是精力用尽了。
自从视界升级到四级、精力上限提升到一千,他已经不怎么需要省着用视界,可以说想开就开,有时忘了关就一直开着。
这还是他第一次把精力用尽。
韩路一默默回想:开始做适配前的精力值是多少来着?
六百多?
他低头看了一下时间,还不到六点。
也就是说,不到一个小时,他已经把可能要耗费大型工程团队几个月时间的工作做完了,同时消耗掉六百多点精力。
这么看来,今晚没办法继续肝了;但在视界如此给力的情况下,只要规划好精力分配,一个月内把所有算子适配都做完完全可行。
韩路一满意地点了点头,手动在电脑上敲上提交命令。
git add .&& git commit -m "FusedAttentionBackward adapter completed"&& git push
回车。
下班。
……
在韩路一下楼离开办公室的同时,十三楼的人工智能基础设施组还有不少人在加班。
江松然今天开组会时已把接下来的计划传达过了:全力做L100的推理适配。
全组压力都很大。
给国产卡做算子适配,这在全国乃至全世界都是最前沿的工作。
这种工作最不敢保证交付时间:都是没人做过的东西,你根本没办法预测会在什么地方冒出个填不平的大坑。
但江松然说话的语气很笃定。虽然没说明确时间线,大家都感觉到了他的决心。
任务分配到人,每个人负责几个算子,按优先级从高到低做;只要求功能实现,先不做性能调优,做完就提交评审,然后再做下一个。
工作量很大,所以不少人都自愿留下来加班。
范小天是其中一个。
源智基础设施团队的高级工程师,薪水拿得不低,但他不是为了钱来的。
他不一样,他是为了理想。
毕竟,一个人如果靠英伟达的股票已经实现了财务自由,接下来的工作不去追求理想,又去追求什么呢?
当然,范小天一般不会把这句话说出来。
因为听起来太欠揍了。
但他确实是组里最资深、最有话语权的人之一。
原因很简单:他以前在英伟达做的就是类似工作——给大模型训练框架里的内核(Kernal)做适配,做性能分析(Profiling)工具。
可也正因为这份履历,他对公司现在的方向不太看好。
太急功近
(本章未完,请点击下一页继续阅读)
『加入书签,方便阅读』