(全球TMT2021年8月25日訊)燧原科技在一年一度的Hot Chips大會上由首席架構(gòu)師劉彥和資深芯片設(shè)計總監(jiān)馮闖一起介紹了第一代云端訓(xùn)練芯片“邃思1.0”的架構(gòu)細(xì)節(jié)。Hot Chips是全球高性能微處理器和集成電路相關(guān)的重要會議之一,芯片行業(yè)巨頭每年都借此機會展示自己公司的最新成果,包括處理器體系結(jié)構(gòu),基礎(chǔ)架構(gòu)計算平臺,內(nèi)存處理等各類技術(shù)。
邃思1.0是燧原科技2019年12月發(fā)布的第一代云端AI訓(xùn)練芯片,采用眾核結(jié)構(gòu),其計算核心采用了燧原科技自研的GCU-CARE計算引擎。整個SOC擁有32個GCU-CARE計算引擎,組成4個計算群組,全面支持常見AI張量數(shù)據(jù)格式(FP32/FP16/BF16, INT8/INT16/INT32),更全面地支撐客戶業(yè)務(wù)。CARE還創(chuàng)新地通過復(fù)用張量核心,用更有效的晶體管效率提供了標(biāo)量、向量、張量以及多種數(shù)據(jù)精度的計算能力。
GCU-DARE數(shù)據(jù)架構(gòu),面向數(shù)據(jù)流優(yōu)化,在數(shù)據(jù)流動中進(jìn)行處理。512GB/s的HBM和200GB/s的GCU-LARE互聯(lián),數(shù)倍于傳統(tǒng)GPU、CPU;強勁的分布式片上共享緩存,提供10TB/s的超大帶寬;可編程共享緩存,可控線程內(nèi)、線程間數(shù)據(jù)常駐共享,消除不必要的IO訪問,既降低了數(shù)據(jù)訪問延時,又節(jié)約了寶貴的IO帶寬;同時,DARE架構(gòu)還提供數(shù)據(jù)異步加載接口,支持?jǐn)?shù)據(jù)與運算的流水執(zhí)行,提高運算并行度。
四路 GCU-LARE智能互聯(lián),200GB/s的高速低延時片間互聯(lián)接口,靈活支持不同規(guī)模的計算需求,可支持千卡級規(guī)模集群,為大中小型數(shù)據(jù)中心提供基于不同需求的人工智能訓(xùn)練產(chǎn)品組合。
邃思1.0人工智能加速芯片專為云端訓(xùn)練場景設(shè)計,支持CNN、RNN、LSTM、BERT等常用人工訓(xùn)練模型,可用于圖像、流數(shù)據(jù)、語音等訓(xùn)練場景。采用標(biāo)準(zhǔn)PCIe 4.0接口,廣泛兼容主流AI服務(wù)器,可滿足數(shù)據(jù)中心大規(guī)模部署的需求,且能效比領(lǐng)先。
演講的最后部分,劉彥還介紹了上個月剛剛在世界人工智能大會上發(fā)布的“邃思2.0”訓(xùn)練芯片。經(jīng)過全新升級迭代后,邃思2.0的計算能力、存儲和帶寬、互聯(lián)能力較第一代訓(xùn)練產(chǎn)品有巨大提升,對超大規(guī)模的模型支持能力獲得顯著增強。由此,燧原科技成為國內(nèi)首家發(fā)布第二代人工智能訓(xùn)練產(chǎn)品組合的公司。
邃思2.0進(jìn)行了大規(guī)模的架構(gòu)升級,針對人工智能計算的特性進(jìn)行深度優(yōu)化,夯實了支持通用異構(gòu)計算的基礎(chǔ);支持全面的計算精度,涵蓋從FP32、TF32、FP16、BF16到INT8,單精度FP32峰值算力達(dá)到40 TFLOPS,單精度張量TF32峰值算力達(dá)到160 TFLOPS。同時搭載了4顆HBM2E片上存儲芯片,高配支持64 GB內(nèi)存,帶寬達(dá)1.8 TB/s。GCU-LARE也全面升級,提供雙向300 GB/s互聯(lián)帶寬,支持?jǐn)?shù)千張云燧CloudBlazer加速卡互聯(lián),實現(xiàn)優(yōu)異的線性加速比。
而同步升級的馭算TopsRider軟件平臺,成為燧原科技構(gòu)建原始創(chuàng)新軟件生態(tài)的基石。通過軟硬件協(xié)同架構(gòu)設(shè)計,充分發(fā)揮邃思2.0的性能;基于算子泛化技術(shù)及圖優(yōu)化策略,支持主流深度學(xué)習(xí)框架下的各類模型訓(xùn)練;利用Horovod分布式訓(xùn)練框架與GCU-LARE互聯(lián)技術(shù)相互配合,為超大規(guī)模集群的高效運行提供解決方案。開放升級的編程模型和可擴展的算子接口,為客戶模型的優(yōu)化提供了自定義的開發(fā)能力。