torchrec适配npu说明

代码结构

.
├── build_whl_torchrec1.2.0.sh   # torchrec==1.2.0+npu构建脚本
├── torchrec1.2.0_npu.patch      # 基于torchrec1.2.0,适配NPU修改后的patch文件
└── README.md                    # 适配说明

说明

1.本样例提供了一个简洁的torchrec适配到昇腾设备方案，用户可快速编译、安装torchrec的npu适配版本。

2.本样例基于开源torchrec v1.2.0版本进行适配。配套开源软件版本：torch 2.7.1, fbgemm_gpu 1.2.0+cpu。

编译安装

基于torchrec v1.2.0版本

# 在当前目录下载源码并编译
git clone -b release/v1.2.0 https://github.com/pytorch/torchrec.git
cd torchrec && git checkout 5db1a21 && cd ..
bash build_whl_torchrec1.2.0.sh
# 安装
cd torchrec/dist
pip3 install torchrec-1.2.0+npu-*.whl
# 安装torchrec依赖。若提示安装expecttest失败，可忽略，该包仅测试场景使用
pip3 install -r requirements.txt

说明

若环境无网络，采用“手动下载torchrec源码并上传到目标环境”的方式获取源码时，需注意：

1. 下载源码包时需切换分支（示例：release/v1.2.0，release/v1.2.0）后再下载（patch文件是基于指定分支修改，不适配torchrec仓库默认的master分支）。
2. 在目标环境上解压缩后，需将解压出来的文件夹名称改为torchrec，使文件夹名称和上述指令及build_whl.sh脚本中的文件夹名称一致。