面向MT-3000异构处理器的PyTorch虚拟算子设计与实现
南开大学学报(自然科学版)
页数： 5 2023-12-20
摘要： 国防科技大学团队自主研发设计了MT-3000高性能多区异构处理器,并将其作为重要算力部署于我国新一代天河超级计算机系统中.为拓展MT-3000在人工智能领域的应用前景,亟需攻克Py Torch等主流深度学习框架仅与国外少数厂商的硬件深度绑定这一技术壁垒.聚焦于Py Torch中常用算子与MT-3000异构处理器适配方法的研究.针对MT-3000处理器中,部分Py Torch算子... （共5页）
APP 