fix #381: fix a bug in PointToVoxel

b0ff62f3 · yan.yan · ab5e2e8e · b0ff62f3 · b0ff62f3 · b0ff62f3
Commit b0ff62f3 authored Nov 19, 2021 by yan.yan
Show whitespace changes
Inline Side-by-side

Showing with 9 additions and 2 deletions

CHANGELOG.md CHANGELOG.md +8 -0

spconv/pytorch/utils.py spconv/pytorch/utils.py +0 -1

version.txt version.txt +1 -1

No files found.
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
 # 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
 ### Added
 - Add support for pytorch 1.5

--- a/spconv/pytorch/utils.py
+++ b/spconv/pytorch/utils.py
@@ -104,7 +104,6 @@ class PointToVoxel(object):
                num_voxels = res[0].shape[0]
            else:
                pc_tv = torch_tensor_to_tv(pc)
-                stream = get_current_stream()
                voxels_tv = torch_tensor_to_tv(self.voxels)
                indices_tv = torch_tensor_to_tv(self.indices)
                num_per_voxel_tv = torch_tensor_to_tv(self.num_per_voxel)

--- a/version.txt
+++ b/version.txt
-2.1.9
+2.1.10