昇腾社区首页
中文
注册

软件架构

图1 软件架构图

Rec SDK Torch基于TorchRec、推荐场景主流框架、CANN和各种硬件和网络,对于搜索、推荐、广告模型训练的应用场景需求,提供极简易用、高性能API,助力昇腾AI处理器完成搜索、推荐、广告等模型的高效训练。

表1 结构图模块介绍

Rec SDK Torch模块

说明

推荐接口层

提供易用性接口、简化用户接入和迁移成本。支持用户规模化上量。

推荐功能层

核心功能实现层,满足用户的使用要求。

推荐加速层

性能竞争力核心组件,为整机系统提供更优性能。

推荐存储层

支持稀疏表的分布式存储。

Torch-npu

开源Torch的昇腾适配层。