MinkowskiEngine icon indicating copy to clipboard operation
MinkowskiEngine copied to clipboard

Fix: stride and coord on different devices

Open Gofinge opened this issue 2 years ago • 1 comments

Fix the bug min_coords and tensor_stride on different devices and raise an error on line 414. A better solution is to assign value to self.device even if it is None in self.init? Then we can make all new tensors can be generated on the specific device directly, instead of .to().

Gofinge avatar May 28 '22 13:05 Gofinge

More specific descriptions of this bug: Assume self.C and self.F are on 'cuda' -> 'min_coord = min(self.C) is on 'cuda' -> tensor_stride = torch.IntTensor() is on 'cpu' -> Boom.

Gofinge avatar May 28 '22 13:05 Gofinge