README_hip.md 523 Bytes
Newer Older
limm's avatar
limm committed
1
2
3
4
5
6
7
8
9
10
11
12
# PyTorch Sparse

## 编译问题

### 编译torch_sparse==0.6.15 目前不支持torch2.1版本的torch,如果编译torch2.1版本的torch_sparse0.6.15版本需要修改一些代码

### 文件路径pytorch_sparse/csrc/version.cpp

```
static auto registry = torch::RegisterOperators().op("torch_sparse::cuda_version", &sparse::cuda_version); -> static auto registry = torch::RegisterOperators().op("torch_sparse::cuda_version", [] { return sparse::cuda_version(); });
```
# 注释:上述内容为README.md的补充。