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