"torchvision/transforms/_functional_tensor.py" did not exist on "21deb4d0f78c47272c1280a32e83c4fd16013997"
Unverified Commit 9802d411 authored by Wenwei Zhang's avatar Wenwei Zhang Committed by GitHub
Browse files

Fix Pytorch 1.8 Compilation issue (#404)

* fix pt1.8 issues

* fix pt1.8 issues
parent e3813185
......@@ -38,7 +38,7 @@ jobs:
strategy:
matrix:
python-version: [3.6, 3.7]
torch: [1.5.0+cu101, 1.6.0+cu101, 1.7.0+cu101]
torch: [1.5.0+cu101, 1.6.0+cu101, 1.7.0+cu101, 1.8.0+cu101]
include:
- torch: 1.5.0+cu101
torchvision: 0.6.0+cu101
......@@ -52,6 +52,10 @@ jobs:
mmcv: 1.7.0+cu101
torchvision: 0.8.1+cu101
cuda_arch: "7.0"
- torch: 1.8.0+cu101
mmcv: 1.8.0+cu101
torchvision: 0.9.0+cu101
cuda_arch: "7.0"
steps:
- uses: actions/checkout@v2
......
......@@ -269,7 +269,7 @@ std::vector<at::Tensor> dynamic_point_to_voxel_forward_gpu(
AT_CUDA_CHECK(cudaGetLastError());
coors_map_sorted = coors_map_sorted.cumsum(0, torch::kInt32);
coors_map.index_put_(coors_id_argsort, coors_map_sorted);
coors_map.index_put_({coors_id_argsort}, coors_map_sorted);
const int num_coors_cpu =
coors_map_sorted[-1].cpu().data_ptr<int32_t>()[0] + 1;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment