面向MT-3000異構(gòu)處理器的PyTorch虛擬算子設(shè)計(jì)與實(shí)現(xiàn)
南開大學(xué)學(xué)報(bào)(自然科學(xué)版)
頁數(shù): 5 2023-12-20
摘要: 國防科技大學(xué)團(tuán)隊(duì)自主研發(fā)設(shè)計(jì)了MT-3000高性能多區(qū)異構(gòu)處理器,并將其作為重要算力部署于我國新一代天河超級(jí)計(jì)算機(jī)系統(tǒng)中.為拓展MT-3000在人工智能領(lǐng)域的應(yīng)用前景,亟需攻克Py Torch等主流深度學(xué)習(xí)框架僅與國外少數(shù)廠商的硬件深度綁定這一技術(shù)壁壘.聚焦于Py Torch中常用算子與MT-3000異構(gòu)處理器適配方法的研究.針對(duì)MT-3000處理器中,部分Py Torch算子...