Skip to content

GitLab

Commit b0ff62f3 authored by yan.yan's avatar yan.yan
Browse files

fix #381: fix a bug in PointToVoxel

parent ab5e2e8e
Show whitespace changes
Inline Side-by-side
Showing with 9 additions and 2 deletions
CHANGELOG.md
View file @ b0ff62f3
# Changelog # Changelog
## [2.1.10] - 2021-11-19
### Fixed
- Fixed a bug in utils.PointToVoxel, shouldn't get cuda stream in cpu code
## [2.1.9] - 2021-11-18
### Removed
- Remove a wrong assert
## [2.1.8] - 2021-11-15 ## [2.1.8] - 2021-11-15
### Added ### Added
- Add support for pytorch 1.5 - Add support for pytorch 1.5
......
spconv/pytorch/utils.py
View file @ b0ff62f3
...@@ -104,7 +104,6 @@ class PointToVoxel(object): ...@@ -104,7 +104,6 @@ class PointToVoxel(object):
num_voxels = res[0].shape[0] num_voxels = res[0].shape[0]
else: else:
pc_tv = torch_tensor_to_tv(pc) pc_tv = torch_tensor_to_tv(pc)
stream = get_current_stream()
voxels_tv = torch_tensor_to_tv(self.voxels) voxels_tv = torch_tensor_to_tv(self.voxels)
indices_tv = torch_tensor_to_tv(self.indices) indices_tv = torch_tensor_to_tv(self.indices)
num_per_voxel_tv = torch_tensor_to_tv(self.num_per_voxel) num_per_voxel_tv = torch_tensor_to_tv(self.num_per_voxel)
......
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