面向MT-3000異構處理器的PyTorch虛擬算子設計與實現(xiàn)

摘要： 國防科技大學團隊自主研發(fā)設計了MT-3000高性能多區(qū)異構處理器,并將其作為重要算力部署于我國新一代天河超級計算機系統(tǒng)中.為拓展MT-3000在人工智能領域的應用前景,亟需攻克Py Torch等主流深度學習框架僅與國外少數(shù)廠商的硬件深度綁定這一技術壁壘.聚焦于Py Torch中常用算子與MT-3000異構處理器適配方法的研究.針對MT-3000處理器中,部分Py Torch算子... （共5頁）