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