首富从AI浪潮开始 第328节

　　简单、简洁、优雅，没有任何注释，但是代码本身就能让你看明白它在干什么。

　　就是这种看得懂反而让范小天看不懂了。

　　块内重计算的策略他认识，这是FlashAttention的核心思想之一，用重计算换显存，理论上可以把显存读写量降低到接近理论的最优值。

　　但问题在于，把这个策略移植到L100上可不是一件容易的事。

　　因为L100的共享内存布局和英伟达家的卡不一样，英伟达的实现没办法直接搬过来用，必须重新写一套专门适配L100内存架构的版本。

　　这个版本要怎么写？得靠真实的性能分析（Profiling）数据反复校准。

　　可是我们现在连性能分析工具（Profiler）都还没写出来呢。

　　这就相当于在没有仪表盘、没有导航，甚至没有地图的道路上赛车，还要排水渠过弯，把每一个弯道都跑出理论极限来。

　　这怎么可能呢？藤原拓海来的也做不到吧。

　　所谓外行看热闹，内行看门道。

　　在英伟达干了这么多年，范小天太清楚这个部分有多难写了。

　　这可不是光懂内存管理的原理就够的，你还得对这块芯片的实际行为真的研究透了才行。因为芯片到了纳米级别，理论和现实之间的偏差只有跑真实数据才能量化。

　　这每一样，单独拿出来，都是个大项目。

　　说白了，这是个得要软硬件都有人懂，还得有人专门研究过L100的专家团队，坐在一起才能做出来的活儿。

　　怎么突然一份完整的代码就出现在代码库里了？

　　这代码——不会没法用吧？

　　抱着试一试的心态，范小天把刚才提交的这份代码从代码库里拉（pull）到本地，建了个测试环境，开始跑正确性验证。

　　他盯着跑起来的进度条，一边打开外卖软件看了一下，一边漫不经心地等着第一个报错弹出来。

　　呀，外卖已经送到了半天了，我的面要坨了。

　　等一会报错了就先去把外卖拿回来。

　　范小天这么想着，抬起头来看向屏幕。

　　他只看到了满屏绿色的PASS。

　　这一套验证测试有几百个，一个一个跑过去，不算快，也不算慢。

　　一行一行的PASS还在日志里输出。

　　PASS。

　　PASS。

　　……

　　PASS。

　　范小天就这么盯着屏幕又看了十分钟，所有的测试都跑完了，全都通过了。

　　最后一行日志：

　　All tests passed 414/414。

　　范小天又把外卖的事情忘在脑后了，他现在满脑子想的都是一件事。

　　这个实现——

　　效率怎么样？

　　因为没有性能分析工具，他没办法读取L100显卡上的各项指标，像SM（Streaming Multiprocessor，流式多处理器）利用率、显存读写效率这些东西，也就没法计算这个实现的运算效率到底有多少。

　　但是还有个取巧的法子，也是源智团队一直在用的。

　　读不了指标没关系，可以直接计时运算时间，再和英伟达的卡比一比，看看能达到竞品卡的多少效率就可以了。

　　按说这个阶段还没到测效率的时候，但是看着这么优雅的代码，范小天忍不住了。

　　万一呢？是吧？

　　范小天迅速搭了个简单的计时基准：同样的Batch大小，同样的序列长度，同样的运算，分别在手头的L100和一张组里共享的H100上跑，计时，对比。

　　这张H100可是传家宝，也不知道江工是从哪里搞来的，组里一起用来做性能测试的，大家都宝贝得不得了。

　　参数设好，范小天看着提交键咽了口口水，才按了下去。

　　进度条跑起来，这次等的时间稍微长一点点，因为性能测试要连续测好几轮，取平均值，这是为了保证测试的稳定性。

　　大概三分钟。

　　结果出来了。

　　L100平均每步用时4.6毫秒。

　　对标的英伟达H100平均每步用时4.8毫秒。

　　比英伟达还快！

　　比英伟达的H100还快！

　　范小天把脸凑近了屏幕，又看了一遍。

　　他以为自己看错了。

　　他又往后退了退，眯着眼睛，又看了一遍。

　　百分之一百零四点二。

　　之前源智团队一直是把运算效率达到了H100的百分之多少作为性能测试的基准的。

　　但是这次是L100直接超过了H100，超了百分之四点二！

　　不对。

　　不对不对不对。

　　范小天把测试工具调出来，准备再跑一遍。

　　要是刚才那遍是跑错了怎么办？

　　又是三分钟。

　　结果和刚才一样。

　　L100在【注意力融合反向算子】上的单步运算速度稳定的超过了H100。

　　范小天抬起头，眼睛里露出迷茫的神色。

　　这个世界是怎么了？

　　我不是回来追求理想，报效祖国，突破万恶的美弟国主义算力封锁的吗？

　　怎么突然一下国家就不需要我的帮助了？

　　怎么国产卡已经算力突破了？

　　范小天在椅子上坐了很久，没有动。

　　这时，模型组的小牛从外面进来，手里提着一个外卖盒。

　　“范哥，这是你的外卖啊？我看放在门口，帮你带进来了，都凉了。”小牛把外卖盒放在范小天的桌子上，准备回工位加班了。

　　小牛还正在为结丹而努力呢，今天下午猛猛肝了一个任务，一个新的成长事件已经在“深蓝”里弹出来了，他相信这么下去，他离金丹期就不远了，肯定能比小杨先进阶。

　　可以可以，金丹在望，元婴可期，我就是源智的化神老祖。

　　这时，范小天才回过神来，他抬头看向小牛，问了一个问题。

　　“诶，你知道谁的账号是null-pointer吗？”

　　小牛用奇怪的眼神看了一眼范小天：“你不知道？”

　　范小天一愣，怎么，我应该知道吗？

　　“这是谁啊？”

　　小牛用手指了指天上，开口说道：

　　“那是韩总的账号啊。”

? 第二百八十九章 黄老板的游说

　　小牛说完这句话，看到范小天的表情，还以为自己说错了什么。

　　范小天坐在椅子上，整个人都僵住了。

　　过了好一会儿，他才开口问道：“韩总？”

　　哪个韩总？

　　大老板韩总？

　　“对啊。”小牛点点头，“韩总在成立源码之前就是个技术大神了，在技术论坛里很出名的？NullPointer是他的网名，空指针，很程序员吧。”

　　范小天看着屏幕上那个提交者用户名。

　　[email protected]